diff --git a/Dockerfile b/Dockerfile
index bbb9d32e..800357ab 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,7 +1,3 @@
-LABEL org.opencontainers.image.source=https://github.com/codelion/optillm
-LABEL org.opencontainers.image.description="OptiLLM full image with model serving and API routing capabilities"
-LABEL org.opencontainers.image.licenses=Apache-2.0
-
 # Build stage
 FROM python:3.12-slim AS builder
 
@@ -30,6 +26,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Final stage
 FROM python:3.12-slim
 
+# Add labels for the final image
+LABEL org.opencontainers.image.source=https://github.com/codelion/optillm
+LABEL org.opencontainers.image.description="OptiLLM full image with model serving and API routing 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 && \
diff --git a/setup.py b/setup.py
index 136dea32..2f5fe03a 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="optillm",
-    version="0.0.32",
+    version="0.0.33",
     packages=find_packages(),
     py_modules=['optillm'],
     package_data={