diff --git a/comfyui/Dockerfile b/comfyui/Dockerfile
index 402a0cf..81a74f9 100644
--- a/comfyui/Dockerfile
+++ b/comfyui/Dockerfile
@@ -1,17 +1,18 @@
 FROM intel/intel-extension-for-pytorch:2.7.10-xpu
 
-# optional, might help with memory allocation performance and scalability
+# Optional, might help with memory allocation performance and scalability
 RUN apt-get update && \
     apt-get install -y --no-install-recommends --fix-missing libjemalloc-dev
 ENV LD_PRELOAD=libjemalloc.so.2
 
-# git clone and run sdnext
+# Download the ComfyUI repository
 RUN echo '#!/bin/bash\ngit status || git clone https://github.com/comfyanonymous/ComfyUI.git /app \npip install -r /app/requirements.txt \npython /app/main.py "$@"' | tee /bin/startup.sh
+
+# Make the startup script executable
 RUN chmod 755 /bin/startup.sh
 
-# actually run sdnext
+# Set the working directory to /app
 WORKDIR /app
-ENTRYPOINT [ "startup.sh", "--highvram", "--use-pytorch-cross-attention", "--listen=0.0.0.0", "--port=8188" ]
 
-# stop signal
-STOPSIGNAL SIGINT
+# Run ComfyUI with custom parameters
+CMD [ "startup.sh", "--highvram", "--use-pytorch-cross-attention", "--listen=0.0.0.0", "--port=8188" ]
diff --git a/sdnext/Dockerfile b/sdnext/Dockerfile
index 8c8638c..ca692b6 100644
--- a/sdnext/Dockerfile
+++ b/sdnext/Dockerfile
@@ -1,16 +1,17 @@
 FROM intel/intel-extension-for-pytorch:2.7.10-xpu
 
-# set paths to use with sdnext
+# Set paths to use with sdnext
 ENV SD_DATADIR="/mnt/data"
 ENV SD_MODELSDIR="/mnt/models"
 
-# git clone and start sdnext
+# Download the SDNext repository
 RUN echo '#!/bin/bash\ngit status || git clone https://github.com/vladmandic/sdnext.git .\npython /app/launch.py "$@"' | tee /bin/startup.sh
+
+# Make the startup script executable
 RUN chmod 755 /bin/startup.sh
 
-# run sdnext
+# Set the working directory to /app
 WORKDIR /app
-ENTRYPOINT [ "startup.sh", "-f", "--use-ipex", "--uv", "--listen", "--debug", "--api-log", "--log", "sdnext.log" ]
 
-# stop signal
-STOPSIGNAL SIGINT
+# Run SDNext with custom parameters
+CMD [ "startup.sh", "-f", "--use-ipex", "--uv", "--listen", "--debug", "--api-log", "--log", "sdnext.log" ]