diff --git a/Dockerfile b/Dockerfile
index 677405b..27c5923 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,9 @@
-ARG PROD=false
-
 FROM node:20.12-slim
 
+# Set the PROD environment variable from the build argument
+ARG PROD
+ENV PROD=$PROD
+
 # Update and install dependencies
 RUN apt-get update && apt-get install -y curl unzip && rm -rf /var/lib/apt/lists/*