diff --git a/.github/workflows/cd-syft.yml b/.github/workflows/cd-syft.yml
index 311e85a4644..6d207e73c84 100644
--- a/.github/workflows/cd-syft.yml
+++ b/.github/workflows/cd-syft.yml
@@ -21,20 +21,24 @@ on:
           - TEST_PYPI
           - REAL_AND_TEST_PYPI
 
+      test_branch:
+        description: "Branch name"
+        required: true
+
 jobs:
   call-pr-tests-linting:
     if: github.repository == 'OpenMined/PySyft' && (github.event.inputs.skip_tests == 'false' || github.event_name == 'schedule') # don't run on forks
-    uses: OpenMined/PySyft/.github/workflows/pr-tests-linting.yml@${{ github.ref }}
+    uses: OpenMined/PySyft/.github/workflows/pr-tests-linting.yml@${{ github.event.inputs.test_branch }}
     secrets: inherit
 
   call-pr-tests-syft:
     if: github.repository == 'OpenMined/PySyft' && (github.event.inputs.skip_tests == 'false' || github.event_name == 'schedule') # don't run on forks
-    uses: OpenMined/PySyft/.github/workflows/pr-tests-syft.yml@${{ github.ref }}
+    uses: OpenMined/PySyft/.github/workflows/pr-tests-syft.yml@${{ github.event.inputs.test_branch }}
     secrets: inherit
 
   call-pr-tests-stack:
     if: github.repository == 'OpenMined/PySyft' && (github.event.inputs.skip_tests == 'false' || github.event_name == 'schedule') # don't run on forks
-    uses: OpenMined/PySyft/.github/workflows/pr-tests-stack.yml@${{ github.ref }}
+    uses: OpenMined/PySyft/.github/workflows/pr-tests-stack.yml@${{ github.event.inputs.test_branch }}
     secrets: inherit
 
   # build-and-push-docker-images: