diff --git a/Dockerfile b/Dockerfile
index 655aa84..6c61afa 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,7 +14,8 @@ RUN npm ci
 COPY . .
 
 # Build the app
-RUN npm run build
+ARG VITE_API_BASE_URL
+RUN VITE_API_BASE_URL=$VITE_API_BASE_URL npm run build
 
 # Stage 2: Static file server (BusyBox)
 FROM busybox:latest