diff --git a/.github/workflows/integration-tests.yml b/.github/workflows/integration-tests.yml
index cadf6d228..19b79b4b7 100644
--- a/.github/workflows/integration-tests.yml
+++ b/.github/workflows/integration-tests.yml
@@ -20,7 +20,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
         with:
-          ref: ${{ github.event.pull_request.head.sha }}
+          ref: ${{ github.event.pull_request.head.ref }}
 
       - shell: bash
         run: |
diff --git a/.github/workflows/test-changes-exists-drift.yml b/.github/workflows/test-changes-exists-drift.yml
index ec474b287..dc529327c 100644
--- a/.github/workflows/test-changes-exists-drift.yml
+++ b/.github/workflows/test-changes-exists-drift.yml
@@ -25,7 +25,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
         with:
-          ref: ${{ github.event.pull_request.head.sha }}
+          ref: ${{ github.event.pull_request.head.ref }}
 
       - shell: bash
         run: |
diff --git a/action.yml b/action.yml
index 886ddbed1..9ada2d528 100644
--- a/action.yml
+++ b/action.yml
@@ -61,6 +61,8 @@ runs:
   steps:
     - name: Checkout
       uses: actions/checkout@v4
+      with:
+        ref: ${{ inputs.sha }}
 
     - name: Set atmos cli config path vars
       shell: bash