diff --git a/.github/workflows/core_contrib_test_0.yml b/.github/workflows/core_contrib_test_0.yml
index a5aea916df..b67b9eae42 100644
--- a/.github/workflows/core_contrib_test_0.yml
+++ b/.github/workflows/core_contrib_test_0.yml
@@ -9,10 +9,12 @@ on:
       CORE_REPO_SHA:
         required: true
         type: string
-
+      CONTRIB_REPO_SHA:
+        required: true
+        type: string
 env:
   CORE_REPO_SHA: ${{ inputs.CORE_REPO_SHA }}
-  CONTRIB_REPO_SHA: main
+  CONTRIB_REPO_SHA: ${{ inputs.CONTRIB_REPO_SHA }}
   PIP_EXISTS_ACTION: w
 
 jobs:
diff --git a/.github/workflows/generate_workflows_lib/src/generate_workflows_lib/core_contrib_test.yml.j2 b/.github/workflows/generate_workflows_lib/src/generate_workflows_lib/core_contrib_test.yml.j2
index 9b0d40f6fd..5553caf8eb 100644
--- a/.github/workflows/generate_workflows_lib/src/generate_workflows_lib/core_contrib_test.yml.j2
+++ b/.github/workflows/generate_workflows_lib/src/generate_workflows_lib/core_contrib_test.yml.j2
@@ -9,10 +9,12 @@ on:
       CORE_REPO_SHA:
         required: true
         type: string
-
+      CONTRIB_REPO_SHA:
+        required: true
+        type: string
 env:
   CORE_REPO_SHA: ${% raw %}{{ inputs.CORE_REPO_SHA }}{% endraw %}
-  CONTRIB_REPO_SHA: main
+  CONTRIB_REPO_SHA: ${% raw %}{{ inputs.CONTRIB_REPO_SHA }}{% endraw %}
   PIP_EXISTS_ACTION: w
 
 jobs: