diff --git a/action.yml b/action.yml
index 8d8d8e9..7cd43f9 100644
--- a/action.yml
+++ b/action.yml
@@ -25,6 +25,10 @@ inputs:
   workingDirectory:
     description: "The working directory in which to run Wrangler"
     required: false
+  wranglerVersion:
+    description: "The version of Wrangler to use"
+    required: false
+    default: 2
 runs:
   using: "node16"
   main: "index.js"
diff --git a/index.js b/index.js
index 93cfa1b..b492455 100644
--- a/index.js
+++ b/index.js
@@ -22069,6 +22069,7 @@ try {
   const gitHubToken = (0, import_core.getInput)("gitHubToken", { required: false });
   const branch = (0, import_core.getInput)("branch", { required: false });
   const workingDirectory = (0, import_core.getInput)("workingDirectory", { required: false });
+  const wranglerVersion = (0, import_core.getInput)("wranglerVersion", { required: false });
   const getProject = async () => {
     const response = await (0, import_undici.fetch)(
       `https://api.cloudflare.com/client/v4/accounts/${accountId}/pages/projects/${projectName}`,
@@ -22084,7 +22085,7 @@ try {
       $ export CLOUDFLARE_ACCOUNT_ID="${accountId}"
     }
   
-    $$ npx wrangler@2 pages publish "${directory}" --project-name="${projectName}" --branch="${branch}"
+    $$ npx wrangler@${wranglerVersion} pages publish "${directory}" --project-name="${projectName}" --branch="${branch}"
     `;
     const response = await (0, import_undici.fetch)(
       `https://api.cloudflare.com/client/v4/accounts/${accountId}/pages/projects/${projectName}/deployments`,
diff --git a/src/index.ts b/src/index.ts
index aea5add..2848443 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -16,6 +16,7 @@ try {
 	const gitHubToken = getInput("gitHubToken", { required: false });
 	const branch = getInput("branch", { required: false });
 	const workingDirectory = getInput("workingDirectory", { required: false });
+	const wranglerVersion = getInput("wranglerVersion", { required: false });
 
 	const getProject = async () => {
 		const response = await fetch(
@@ -34,7 +35,7 @@ try {
       $ export CLOUDFLARE_ACCOUNT_ID="${accountId}"
     }
   
-    $$ npx wrangler@2 pages publish "${directory}" --project-name="${projectName}" --branch="${branch}"
+    $$ npx wrangler@${wranglerVersion} pages publish "${directory}" --project-name="${projectName}" --branch="${branch}"
     `;
 
 		const response = await fetch(