diff --git a/build_scripts/Dockerfile.comfyui.from_scratch b/build_scripts/Dockerfile.comfyui.from_scratch
index ae440e83..fd5d4993 100644
--- a/build_scripts/Dockerfile.comfyui.from_scratch
+++ b/build_scripts/Dockerfile.comfyui.from_scratch
@@ -27,5 +27,5 @@ RUN echo "/opt/ml/code" > "/opt/conda/lib/python3.10/site-packages/packages.pth"
 WORKDIR /opt/ml/code
 
 ENV ON_DOCKER true
-
-ENTRYPOINT ["python", "/opt/ml/code/main.py"]
+ENV SAGEMAKER_PROGRAM /opt/ml/code/main.py
+#ENTRYPOINT ["python", "/opt/ml/code/main.py"]