diff --git a/docker/docker-compose-dev.yaml b/docker/docker-compose-dev.yaml
index 1be103243..b005f74ee 100644
--- a/docker/docker-compose-dev.yaml
+++ b/docker/docker-compose-dev.yaml
@@ -42,6 +42,7 @@ services:
     ports:
       - ${AI_SERVICE_FORWARD_PORT}:${WREN_AI_SERVICE_PORT}
     environment:
+      WREN_AI_SERVICE_PORT: ${WREN_AI_SERVICE_PORT}
       WREN_UI_ENDPOINT: ${WREN_UI_ENDPOINT}
       OPENAI_API_KEY: ${OPENAI_API_KEY}
       GENERATION_MODEL: ${GENERATION_MODEL}