diff --git a/Dockerfile b/Dockerfile
index 6e75e75e..c971c984 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Build stage
-FROM python:3.12-slim-bullseye AS builder
+FROM python:3.12-slim-bookworm AS builder
 
 # Define build argument with default value
 ARG PORT=8000
@@ -26,7 +26,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Final stage
-FROM python:3.12-slim-bullseye
+FROM python:3.12-slim-bookworm
 
 # Add labels for the final image
 LABEL org.opencontainers.image.source="https://github.com/codelion/optillm"
diff --git a/Dockerfile.proxy_only b/Dockerfile.proxy_only
index cc409a1c..a3c9f29a 100644
--- a/Dockerfile.proxy_only
+++ b/Dockerfile.proxy_only
@@ -1,5 +1,5 @@
 # Build stage
-FROM python:3.12-slim-bullseye AS builder
+FROM python:3.12-slim-bookworm AS builder
 
 # Define build argument with default value
 ARG PORT=8000
@@ -26,7 +26,7 @@ COPY requirements_proxy_only.txt .
 RUN pip install --no-cache-dir -r requirements_proxy_only.txt
 
 # Final stage
-FROM python:3.12-slim-bullseye
+FROM python:3.12-slim-bookworm
 
 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"
diff --git a/optillm/__init__.py b/optillm/__init__.py
index 9aece42c..9ff9e0a4 100644
--- a/optillm/__init__.py
+++ b/optillm/__init__.py
@@ -2,7 +2,7 @@
 import os
 
 # Version information
-__version__ = "0.1.5"
+__version__ = "0.1.6"
 
 # Get the path to the root optillm.py
 spec = util.spec_from_file_location(
diff --git a/setup.py b/setup.py
index 803b0c3e..2ebd1f13 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name="optillm",
-    version="0.1.5",
+    version="0.1.6",
     packages=find_packages(include=['optillm', 'optillm.*']),  # This ensures all subpackages are included
     py_modules=['optillm'],
     package_data={