diff --git a/.github/workflows/pypi_deployment.yml b/.github/workflows/pypi_deployment.yml
index 249f700..0add5aa 100644
--- a/.github/workflows/pypi_deployment.yml
+++ b/.github/workflows/pypi_deployment.yml
@@ -91,7 +91,7 @@ jobs:
 
     - uses: actions/checkout@v4
       with:
-        ref: 'main'
+        ref: "dev-${{ github.ref_name.split('v')[1] }}"
 
     - name: Install dependencies
       run: |