diff --git a/.github/workflows/test-action.yml b/.github/workflows/test-action.yml
index d39796a..2cdfd0e 100644
--- a/.github/workflows/test-action.yml
+++ b/.github/workflows/test-action.yml
@@ -54,11 +54,12 @@ jobs:
 
        - name: Build image
          id: build
-         uses: usimd/pi-gen-action@v1
+         uses: usimd/pi-gen-action@main
          with:
            image-name: raspios
            compression: none
-           pi-gen-version: armhf
+           pi-gen-version: master
+           verbose-output: true
            stage-list: stage0 stage1 stage2 ./test-stage
 
        - name: Build image