diff --git a/dist/index.js b/dist/index.js
index 3100f0a3..887c39aa 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -146,8 +146,15 @@ function computeVersion(version, repoToken) {
     });
 }
 function getFileName() {
+    var _a;
     const platform = osPlat === "win32" ? "windows" : osPlat;
-    const arch = osArch === "x64" ? "amd64" : "386";
+    const arches = {
+        arm: "arm",
+        arm64: "arm64",
+        x64: "amd64",
+        ia32: "386",
+    };
+    const arch = (_a = arches[osArch]) !== null && _a !== void 0 ? _a : osArch;
     const ext = osPlat === "win32" ? "zip" : "tar.gz";
     const filename = util.format("task_%s_%s.%s", platform, arch, ext);
     return filename;
diff --git a/src/installer.ts b/src/installer.ts
index d4c1a989..34f846b9 100644
--- a/src/installer.ts
+++ b/src/installer.ts
@@ -136,7 +136,13 @@ async function computeVersion(
 
 function getFileName() {
   const platform: string = osPlat === "win32" ? "windows" : osPlat;
-  const arch: string = osArch === "x64" ? "amd64" : "386";
+  const arches = {
+    arm: "arm",
+    arm64: "arm64",
+    x64: "amd64",
+    ia32: "386",
+  };
+  const arch: string = arches[osArch] ?? osArch;
   const ext: string = osPlat === "win32" ? "zip" : "tar.gz";
   const filename: string = util.format("task_%s_%s.%s", platform, arch, ext);