diff --git a/services/comfy/Dockerfile b/services/comfy/Dockerfile
index ec63220..3984675 100644
--- a/services/comfy/Dockerfile
+++ b/services/comfy/Dockerfile
@@ -41,7 +41,6 @@ RUN if [ "$PUID" -ne "0" ]; then \
       echo "root group detected" ; \
     fi
 
-RUN chown -R "$PUID:$PGID" /stable-diffusion
 USER $PUID:$PGID
 
 ENV ROOT=/stable-diffusion
@@ -52,6 +51,8 @@ RUN --mount=type=cache,target="$USER_HOME"/.cache/pip \
   git reset --hard 276f8fce9f5a80b500947fb5745a4dde9e84622d && \
   pip install -r requirements.txt
 
+RUN chown -R "$PUID:$PGID" "${ROOT}"
+
 WORKDIR ${ROOT}
 COPY --chown=$PUID:$PGID . /docker
 RUN chmod u+x /docker/entrypoint.sh && cp /docker/extra_model_paths.yaml ${ROOT}