diff --git a/Dockerfile b/Dockerfile
index 7dd1387..2419550 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,7 @@
 FROM pytorch/pytorch:latest
 
 ENV NVIDIA_VISIBLE_DEVICES=all
-ENV CLI_ARGS=""
+ENV PYTHONPATH="${PYTHONPATH}:${PWD}" CLI_ARGS=""
 
 RUN apt update && apt install git python3-pip libsm6 libxext6 -y && apt clean
 
@@ -10,9 +10,10 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
 	cd ${ROOT} && \
 	pip install -r requirements.txt
 
-RUN cd ${ROOT}/custom_nodes && \
+WORKDIR ${ROOT}
+
+RUN cd ./custom_nodes && \
     git clone https://github.com/ltdrdata/ComfyUI-Manager.git
 
-WORKDIR ${ROOT}
 EXPOSE 5555
 CMD python -u main.py --listen --port 5555 ${CLI_ARGS}
\ No newline at end of file