diff --git a/action.yaml b/action.yaml
index 99ade7523d..0a655e53bd 100644
--- a/action.yaml
+++ b/action.yaml
@@ -144,7 +144,7 @@ runs:
 
     - name: Setup swapfile
       shell: bash
-      if: matrix.os == 'ubuntu-24.04'
+      if: runner.arch == 'X64' || runner.arch == 'X86'
       run: |
         sudo fallocate -l 16G /swapfile
         sudo chmod 600 /swapfile