diff --git a/index.js b/index.js
index a06c444..83e8637 100644
--- a/index.js
+++ b/index.js
@@ -16124,7 +16124,7 @@ try {
       $ export CLOUDFLARE_ACCOUNT_ID="${accountId}"
     }
 
-    if ${environment} {
+    if ${branch} {
       $$ npx wrangler@2 pages publish "${directory}" --project-name="${projectName}" --branch="${branch}"
     } else {
       $$ npx wrangler@2 pages publish "${directory}" --project-name="${projectName}"
diff --git a/src/index.ts b/src/index.ts
index ff40235..7ab720e 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -65,7 +65,7 @@ try {
       $ export CLOUDFLARE_ACCOUNT_ID="${accountId}"
     }
 
-    if ${environment} {
+    if ${branch} {
       $$ npx wrangler@2 pages publish "${directory}" --project-name="${projectName}" --branch="${branch}"
     } else {
       $$ npx wrangler@2 pages publish "${directory}" --project-name="${projectName}"