diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 8ebca4d..2e6f867 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -17,5 +17,6 @@ jobs:
       - uses: viamrobotics/build-action@v1
         with:
           version: ""  # Empty = build only, no deploy
+          ref: ${{ github.event.pull_request.head.sha || github.sha }}
           key-id: ${{ secrets.VIAM_DEV_API_KEY_ID }}
           key-value: ${{ secrets.VIAM_DEV_API_KEY }}