diff --git a/docker-compose.yml b/docker-compose.yml
index 37352a3..d379f25 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -2,7 +2,7 @@ version: "3.8"
 
 services:
   comfyui:
-    image: timpietruskyblibla/runpod-worker-comfy:latest
+    image: timpietruskyblibla/runpod-worker-comfy:dev
     container_name: comfyui-worker
     environment:
       - NVIDIA_VISIBLE_DEVICES=all
@@ -12,3 +12,4 @@ services:
     runtime: nvidia
     volumes:
       - ./data/comfyui/output:/comfyui/output
+      - ./data/runpod-volume:/runpod-volume