diff --git a/.github/workflows/image-update.yaml b/.github/workflows/image-update.yaml
index 082b127..c660bd4 100644
--- a/.github/workflows/image-update.yaml
+++ b/.github/workflows/image-update.yaml
@@ -17,6 +17,8 @@ jobs:
     steps:
       - name: Checkout repository
         uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
+        with:
+          ref: ${{ github.head_ref }}
 
       - name: Install Zarf
         uses: defenseunicorns/setup-zarf@main
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index 04d9b04..e6908ed 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -27,8 +27,6 @@ jobs:
     steps:
       - name: Checkout repository
         uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
-        with:
-          ref: ${{ github.head_ref }}
 
       - name: Environment setup
         uses: ./.github/actions/setup