diff --git a/Dockerfile b/Dockerfile
index 70041f41d..73166b96d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,7 +35,9 @@ RUN pnpm install && \
     pnpm build
 
 # Give the node user ownership of the application files
-RUN chown -R node:node .
+RUN chown -R node:node . && \
+    mkdir -p /home/node/.flowise && \
+    chown -R node:node /home/node/.flowise
 
 # Switch to non-root user (node user already exists in node:20-alpine)
 USER node
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 2c61336f3..a0c0c7d5f 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -2,7 +2,7 @@ version: '3.1'
 
 services:
     flowise:
-        image: flowiseaiyau/test:latest
+        image: flowiseaiyau/test:0.0.2
         restart: always
         environment:
             - PORT=${PORT}
@@ -142,5 +142,8 @@ services:
             retries: 5
             start_period: 30s
         volumes:
-            - ~/.flowise:/home/node/.flowise
+            - flowise_data:/home/node/.flowise
         entrypoint: /bin/sh -c "sleep 3; pnpm start"
+
+volumes:
+    flowise_data:
\ No newline at end of file