diff --git a/.github/workflows/publish_package.yml b/.github/workflows/publish_package.yml
index df30aac..fd61fd0 100644
--- a/.github/workflows/publish_package.yml
+++ b/.github/workflows/publish_package.yml
@@ -13,10 +13,10 @@ jobs:
     steps:
       - uses: actions/checkout@v3
 
-      - name: Set up Python 3.10
+      - name: Set up Python 3.12
         uses: actions/setup-python@v4
         with:
-          python-version: '3.10'
+          python-version: '3.12'
 
       - name: Install build and tomlkit dependencies
         run: python -m pip install --upgrade pip build tomlkit
@@ -42,7 +42,9 @@ jobs:
 
       - name: Install Comfy CLI from Test Pypi and Test
         run: |
-          pip install -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple comfy-cli==${{env.VERSION}}
+          for i in {1..3}; do
+            pip install -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple comfy-cli==${{env.VERSION}} && break || sleep 5
+          done
           comfy --help
 
       - name: Publish distribution to Official PyPI