diff --git a/.github/actions/vmtest/action.yml b/.github/actions/vmtest/action.yml
index fa607c4..fcd1a78 100644
--- a/.github/actions/vmtest/action.yml
+++ b/.github/actions/vmtest/action.yml
@@ -37,6 +37,7 @@ runs:
       uses: libbpf/ci/setup-build-env@main
       with:
         pahole: ${{ inputs.pahole }}
+        arch: ${{ inputs.arch }}
     # 1. download CHECKPOINT kernel source
     - name: Get checkpoint commit
       shell: bash