diff --git a/.gitea/workflows/build.yaml b/.gitea/workflows/build.yaml
index 00e1743..8624833 100644
--- a/.gitea/workflows/build.yaml
+++ b/.gitea/workflows/build.yaml
@@ -44,18 +44,23 @@ jobs:
           cache-from: type=gha
           cache-to: type=gha,mode=max
 
-      - name: Update RunPod Endpoint
+      - name: Update RunPod Template
         run: |
-          curl -s -X POST "https://api.runpod.io/graphql?api_key=${{ secrets.RUNPOD_API_KEY }}" \
+          curl -s -X POST "https://rest.runpod.io/v1/templates/${{ secrets.RUNPOD_TEMPLATE_ID }}/update" \
+            -H "Authorization: Bearer ${{ secrets.RUNPOD_API_KEY }}" \
             -H "Content-Type: application/json" \
-            -d '{
-              "query": "mutation { saveEndpoint(input: { id: \"${{ secrets.RUNPOD_ENDPOINT_ID }}\", imageName: \"flybynight69420/comfyui-serverless:latest\" }) { id imageName } }"
-            }'
+            -d '{"imageName": "flybynight69420/comfyui-serverless:latest"}'
 
-      - name: Purge RunPod Workers
+      - name: Restart RunPod Workers
         run: |
           curl -s -X POST "https://api.runpod.io/graphql?api_key=${{ secrets.RUNPOD_API_KEY }}" \
             -H "Content-Type: application/json" \
             -d '{
-              "query": "mutation { endpointPurgeQueue(input: { endpointId: \"${{ secrets.RUNPOD_ENDPOINT_ID }}\" }) }"
+              "query": "mutation { saveEndpoint(input: { id: \"${{ secrets.RUNPOD_ENDPOINT_ID }}\", name: \"comfyui-serverless\", workersMin: 0, workersMax: 0 }) { id } }"
+            }'
+          sleep 5
+          curl -s -X POST "https://api.runpod.io/graphql?api_key=${{ secrets.RUNPOD_API_KEY }}" \
+            -H "Content-Type: application/json" \
+            -d '{
+              "query": "mutation { saveEndpoint(input: { id: \"${{ secrets.RUNPOD_ENDPOINT_ID }}\", name: \"comfyui-serverless\", workersMin: 0, workersMax: 1 }) { id } }"
             }'