diff --git a/.github/actions/setup-publish/action.yml b/.github/actions/setup-publish/action.yml
index abcddf28..fa15024c 100644
--- a/.github/actions/setup-publish/action.yml
+++ b/.github/actions/setup-publish/action.yml
@@ -1,19 +1,9 @@
 name: Setup publish
 description: Setup steps for publishing packages
 
-inputs:
-  branch:
-    required: true
-
 runs:
   using: 'composite'
   steps:
-    - name: Checkout repository
-      uses: actions/checkout@v4
-      with:
-        fetch-depth: 0
-        ref: ${{ inputs.branch }}
-
     - name: Install pnpm
       uses: pnpm/action-setup@v4
       with:
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
index c2f612a7..aea9b908 100644
--- a/.github/workflows/publish.yml
+++ b/.github/workflows/publish.yml
@@ -42,10 +42,13 @@ jobs:
       id-token: write # OIDC for provenance if npm publish happens here
     runs-on: ubuntu-latest
     steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
       - name: Setup publish
         uses: ./.github/actions/setup-publish
-        with:
-          branch: ${{ github.ref_name }}
 
       # This action creates a release pull request with all of
       # the package versions and changelogs updated. When there
@@ -112,10 +115,14 @@ jobs:
       id-token: write # OIDC for provenance when npm publish happens
     runs-on: ubuntu-latest
     steps:
+      # Checkout the branch selected when triggering the workflow
+      - name: Checkout repository
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+
       - name: Setup publish
         uses: ./.github/actions/setup-publish
-        with:
-          branch: ${{ github.ref_name }}
 
       - name: Version packages for snapshot
         run: pnpm changeset version --snapshot ${{ inputs.snapshot_tag }}