diff --git a/Dockerfile b/Dockerfile
index 49928e2c17..2e87c63670 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -161,8 +161,11 @@ ENV S3_BUCKET_NAME=request-response-storage
 ENV S3_PROMPT_BUCKET_NAME=prompt-body-storage
 ENV BETTER_AUTH_SECRET=change-me-in-production
 
+# Pull the AI Gateway binary from the published image so supervisord can run it.
+COPY --from=helicone/ai-gateway:latest /usr/local/bin/ai-gateway /usr/local/bin/ai-gateway
+
 CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/conf.d/supervisord.conf"]
 
 # --------------------------------------------------------------------------------------------------------------------
 
-EXPOSE 3000 8585 8123 9080 9001 5432
\ No newline at end of file
+EXPOSE 3000 8585 8123 8788 9080 9001 5432
\ No newline at end of file
diff --git a/supervisord.conf b/supervisord.conf
index 312697e78f..249ef83347 100644
--- a/supervisord.conf
+++ b/supervisord.conf
@@ -92,8 +92,7 @@ stdout_logfile=/var/log/supervisor/minio-setup.out.log
 environment=MINIO_ROOT_USER="minioadmin",MINIO_ROOT_PASSWORD="minioadmin"
 
 [program:ai-gateway]
-command=yarn start
-directory=/app/gateway
+command=/usr/local/bin/ai-gateway
 autostart=true
 autorestart=true
 stderr_logfile=/var/log/supervisor/ai-gateway.err.log