diff --git a/.github/workflows/schemas.yml b/.github/workflows/schemas.yml
index f385cc4..3741ec6 100644
--- a/.github/workflows/schemas.yml
+++ b/.github/workflows/schemas.yml
@@ -16,7 +16,7 @@ on:
     types: [generate-pipeline-schemas]
 
 env:
-  HAYSTACK_REF: ${{ github.event.client_payload.ref || inputs.ref || 'v1.x' }}
+  HAYSTACK_REF: ${{ github.event.client_payload.ref || inputs.ref || 'main' }}
   SLACK_WEBHOOK_URL: ${{ secrets.SLACK_WEBHOOK_URL }}
 
 jobs:
@@ -49,7 +49,7 @@ jobs:
           git config --local user.email "github-actions[bot]@users.noreply.github.com"
           git config --local user.name "github-actions[bot]"
           git add .
-          if [ "${HAYSTACK_REF}" != "v1.x" ]; then
+          if [ "${HAYSTACK_REF}" != "main" ]; then
             git restore --staged --worktree json-schema/haystack-pipeline-main.schema.json
           fi
           git commit -m "Update Schema for ${HAYSTACK_REF}" -a || echo "No changes to commit"