diff --git a/.github/workflows/python_wheel_build.yml b/.github/workflows/python_wheel_build.yml
index bfa6db5eacd91..a3e980f66cc5c 100644
--- a/.github/workflows/python_wheel_build.yml
+++ b/.github/workflows/python_wheel_build.yml
@@ -79,7 +79,7 @@ jobs:
 
   create-and-upload-wheel-registry:
       if: github.event_name != 'pull_request' # The secrets are not available in PR
-      needs: build-wheels
+      needs: test-wheels
       runs-on: ubuntu-latest
       steps:
 
@@ -127,7 +127,7 @@ jobs:
 
   publish-to-pypi:
       if: github.event_name == 'workflow_dispatch'
-      needs: build-wheels
+      needs: test-wheels
       runs-on: ubuntu-latest
       name: >-
         Publish Python distribution to PyPI