diff --git a/docker-compose.yml b/docker-compose.yml
index 63ff166..f35247d 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -29,6 +29,8 @@ services:
     build:
       context: ./frontend
     image: cafeteria-frontend:${FRONTEND_TAG}
+    environment:
+      VITE_API_BASE: ${VITE_API_BASE}
     networks:
       - cafeteria
     ports:
diff --git a/frontend/Dockerfile b/frontend/Dockerfile
index 9ead6b2..58ea81b 100644
--- a/frontend/Dockerfile
+++ b/frontend/Dockerfile
@@ -1,6 +1,8 @@
 FROM node:20-slim AS build
 WORKDIR /app
 COPY package*.json ./
+ARG VITE_API_BASE
+ENV VITE_API_BASE=$VITE_API_BASE
 RUN npm install --no-progress
 COPY . .
 RUN npm run build