diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index 746d4a0988..c4c80ebdcc 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -50,7 +50,7 @@ jobs:
 
   build:
     name: Build distribution 📦
-    needs: trigger-client-build  # Wait for client build to complete
+    needs: trigger-client-and-models-build  # Wait for client build to complete
     runs-on: ubuntu-latest
 
     steps: