diff --git a/dockerfile b/dockerfile
index 2fc97f9..5137385 100644
--- a/dockerfile
+++ b/dockerfile
@@ -43,10 +43,11 @@ WORKDIR /app
 # Pull source and extract
 RUN wget https://github.com/comfyanonymous/ComfyUI/archive/refs/tags/v${ComfyUI_Release}.tar.gz && \
         tar xzf v${ComfyUI_Release}.tar.gz && \
-        rm v${ComfyUI_Release}.tar.gz
+        rm v${ComfyUI_Release}.tar.gz && \
+        mv /app/ComfyUI-${ComfyUI_Release} /app/comfyui/
 
 # Set workdir for application
-WORKDIR /app/ComfyUI-${ComfyUI_Release}
+WORKDIR /app/comfyui
 
 # Upgrade pip and install dependencies
 RUN python -m pip install --upgrade pip && \
@@ -57,4 +58,4 @@ RUN python -m pip install --upgrade pip && \
 EXPOSE 8188
 
 # Start CompyUI
-CMD ["python", "main.py"]
+CMD ["python", "main.py"]
\ No newline at end of file