diff --git a/Dockerfile b/Dockerfile
index 11e1f6e01..e124ca428 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,8 +5,6 @@
 # Use the recommended Python version 3.12, as specified in the README.
 FROM python:3.12.11-bookworm AS comfyui-base
 
-ARG APT_EXTRA_PACKAGES
-
 # Install cmake, which is an indirect installation dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends cmake
 
@@ -72,12 +70,13 @@ VOLUME [ "/data", "/comfyui/.venv", "/comfyui/custom_nodes", "/comfyui/models",
 # dependencies
 USER root
 
-# Install additional system dependencies
-RUN apt-get install -y --no-install-recommends $APT_EXTRA_PACKAGES \
-	&& apt-get clean                                               \
-	&& rm -rf /var/lib/apt/lists/*
-
 # Configure entrypoint
 RUN chmod +x entrypoint.sh
 ENTRYPOINT [ "./entrypoint.sh" ]
 CMD [ "python", "./main.py" ]
+
+# Install additional system dependencies
+ARG APT_EXTRA_PACKAGES
+RUN apt-get install -y --no-install-recommends $APT_EXTRA_PACKAGES \
+	&& apt-get clean                                               \
+	&& rm -rf /var/lib/apt/lists/*