diff --git a/action.yml b/action.yml
index 9a7fb84..f3f494c 100644
--- a/action.yml
+++ b/action.yml
@@ -17,6 +17,10 @@ inputs:
     description: Working directory to run the action.
     required: false
     default: .
+  architecture:
+    description: CPU architecture (x86_64 or aarch64)
+    required: false
+    default: x86_64
 outputs:
   paths:
     description: Paths of test files to run.
@@ -25,7 +29,7 @@ runs:
   using: composite
   steps:
     - run: |
-        curl -L --output split-test https://github.com/mtsmfm/split-test/releases/download/v1.1.0/split-test-x86_64-unknown-linux-gnu
+        curl -L --output split-test https://github.com/mtsmfm/split-test/releases/download/v1.1.0/split-test-${{ inputs.architecture }}-unknown-linux-gnu
         chmod +x split-test
       shell: bash
       working-directory: ${{ inputs.working-directory }}