diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml
index 4766d22..805f31f 100644
--- a/docker-compose.dev.yml
+++ b/docker-compose.dev.yml
@@ -68,7 +68,8 @@ services:
       - NODE_ENV=development
       - API_URL=https://api.lago.dev
       - APP_DOMAIN=https://app.lago.dev
-      - CODEGEN_API=https://api.lago.dev:3000
+      - CODEGEN_API_CUSTOMER_PORTAL=https://api.lago.dev/customer_portal/graphql
+      - CODEGEN_API_GRAPHQL=https://api.lago.dev/api/graphql
       - LAGO_DISABLE_SIGNUP=${LAGO_DISABLE_SIGNUP:-}
       - NANGO_SECRET_KEY=${NANGO_SECRET_KEY:-}
     labels:
diff --git a/docker-compose.yml b/docker-compose.yml
index 1ccdf2b..c8e1aa1 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -101,7 +101,6 @@ services:
     environment:
       - API_URL=${LAGO_API_URL:-http://localhost:3000}
       - APP_ENV=${APP_ENV:-production}
-      - CODEGEN_API=${LAGO_API_URL:-http://localhost:3000}
       - LAGO_DISABLE_SIGNUP=${LAGO_DISABLE_SIGNUP:-false}
       - LAGO_OAUTH_PROXY_URL=https://proxy.getlago.com
       - SENTRY_DSN=${SENTRY_DSN_FRONT}