diff --git a/.github/workflows/integration_tests.yaml b/.github/workflows/integration_tests.yaml
index 6f109f3cb..bcc43f852 100644
--- a/.github/workflows/integration_tests.yaml
+++ b/.github/workflows/integration_tests.yaml
@@ -8,6 +8,11 @@ on:
     secrets:
       RUNPOD_API_KEY:
         required: true
+  workflow_dispatch:
+    inputs:
+      lorax_tag:
+        required: true
+        type: string
 jobs:
   deploy:
     runs-on: ubuntu-latest
diff --git a/tests/create-pod.sh b/tests/create-pod.sh
index 2abc7f962..057b5cbd1 100644
--- a/tests/create-pod.sh
+++ b/tests/create-pod.sh
@@ -1,9 +1,9 @@
 IMAGE_NAME="$1"
 
 runpodctl create pods \
-  --name lorax-tests-new \
+  --name lorax-tests \
   --gpuType "NVIDIA A40" \
-  --imageName "$IMAGE_NAME" \
+  --imageName "ghcr.io/predibase/lorax:$IMAGE_NAME" \
   --containerDiskSize 100 \
   --volumeSize 100 \
   --ports "8080/http" \