diff --git a/Dockerfile.proxy_only b/Dockerfile.proxy_only
index e5db1708..b863ee90 100644
--- a/Dockerfile.proxy_only
+++ b/Dockerfile.proxy_only
@@ -1,7 +1,3 @@
-LABEL org.opencontainers.image.source=https://github.com/codelion/optillm
-LABEL org.opencontainers.image.description="OptiLLM proxy-only image for API routing without model serving capabilities"
-LABEL org.opencontainers.image.licenses=Apache-2.0
-
 # Build stage
 FROM python:3.12-slim AS builder
 
@@ -30,6 +26,10 @@ RUN pip install --no-cache-dir -r requirements_proxy_only.txt
 # Final stage
 FROM python:3.12-slim
 
+LABEL org.opencontainers.image.source=https://github.com/codelion/optillm
+LABEL org.opencontainers.image.description="OptiLLM proxy-only image for API routing without model serving capabilities"
+LABEL org.opencontainers.image.licenses=Apache-2.0
+
 # Install curl for the healthcheck
 RUN apt-get update && apt-get install -y --no-install-recommends \
   curl && \