diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index 4cf62c856..52ebd77d1 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -126,8 +126,9 @@ jobs:
 
       - name: Run integration tests
         uses: ./.github/workflows/integration-tests
+        if: ${{ github.ref == 'refs/heads/main' }}
         with:
-          test_image_tag: ${{ steps.meta.outputs.tags }}
+          test_image_tag: ${{ steps.meta.outputs.tags[0] }}
           use_local_image: true
           github_token: ${{ secrets.GHCR_PAT }}
           huggingface_token: ${{ secrets.HUGGING_FACE_HUB_TOKEN }}