diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index f8d3173dd..27ed04977 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -30,4 +30,4 @@ jobs:
     needs: prepare-variables-job
     uses: ./.github/workflows/setup.yml
     with:
-      tf_version: ${{ needs.prepare-variables-job.outputs.tensorflow_version }}
\ No newline at end of file
+      tf_version: ${{ needs.prepare-variables.outputs.tensorflow_version }}
\ No newline at end of file