diff --git a/.kokoro/build.sh b/.kokoro/build.sh
index 7f4da2b7..d2b0af0c 100755
--- a/.kokoro/build.sh
+++ b/.kokoro/build.sh
@@ -83,7 +83,7 @@ if [[ -n "${_LOUHI_REF_NAME:-}" ]]; then
     PACKAGE_DIR="invoker"
   else
     echo "Unknown tag format: ${_LOUHI_REF_NAME}. Defaulting to invoker."
-    PACKAGE_DIR="invoker"
+    PACKAGE_DIR="functions-framework-api"
   fi
 else
   # Fallback for manual/non-tag builds (e.g. testing)
@@ -93,7 +93,7 @@ else
   elif [[ $KOKORO_JOB_NAME == *"functions-framework-api"* ]]; then
     PACKAGE_DIR="functions-framework-api"
   else
-    PACKAGE_DIR="invoker"
+    PACKAGE_DIR="functions-framework-api"
   fi
 fi