diff --git a/.github/workflows/publish_package.yml b/.github/workflows/publish_package.yml
index 2b20a97..78c405d 100644
--- a/.github/workflows/publish_package.yml
+++ b/.github/workflows/publish_package.yml
@@ -73,7 +73,7 @@ jobs:
 
   test-pip-installation:
     name: Test Comfy CLI Installation via pip
-    needs: build-n-publish  # This job runs after build-n-publish completes successfully
+    needs: build-n-publish-pypi  # This job runs after build-n-publish completes successfully
     runs-on: ubuntu-latest
     steps:
       - name: Set up Python
@@ -85,4 +85,4 @@ jobs:
         run: pip install comfy-cli
 
       - name: Test Comfy CLI Help
-        run: comfy --help
\ No newline at end of file
+        run: comfy --help