diff --git a/higgsfield/static/templates/deploy_action.j2 b/higgsfield/static/templates/deploy_action.j2
index 67e2e0b..02c3dbb 100644
--- a/higgsfield/static/templates/deploy_action.j2
+++ b/higgsfield/static/templates/deploy_action.j2
@@ -19,7 +19,7 @@ jobs:
       uses: actions/checkout@v3
     - name: Set credentials
       run: |
-          pip install git+https://github.com/ml-doom/demo.git@main --quiet
+          pip install higgsfield==0.0.1 --quiet
 
           echo "SSH_KEY='NOTHING'" >> env
           eval "$(jq -r '["key", "user", "port", "hosts"] as $names | . as $json | $names[] as $name | "export \($name | ascii_upcase | @sh)=\($json[$name] | @sh)"' < <(higgsfield ci get-ssh-details))"          
diff --git a/higgsfield/static/templates/experiment_action.j2 b/higgsfield/static/templates/experiment_action.j2
index 305b50c..702a611 100644
--- a/higgsfield/static/templates/experiment_action.j2
+++ b/higgsfield/static/templates/experiment_action.j2
@@ -38,7 +38,7 @@ jobs:
         rm invoker-latest-linux-amd64.tar.gz
     - name: Set Port and Run Name
       run: |
-          pip install git+https://github.com/ml-doom/demo.git@main --quiet
+          pip install higgsfield==0.0.1 --quiet
           
           echo "SSH_KEY='NOTHING'" >> env
           eval "$(jq -r '["key", "user", "port", "hosts"] as $names | . as $json | $names[] as $name | "export \($name | ascii_upcase | @sh)=\($json[$name] | @sh)"' < <(higgsfield ci get-ssh-details))"