diff --git a/Dockerfile b/Dockerfile
index 554e96c..2bc401b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,7 +14,7 @@ RUN npm ci --legacy-peer-deps
 COPY . .
 
 # Build args for environment
-ARG VITE_TLDRAW_WORKER_URL=https://jeffemmett-canvas.jeffemmett.workers.dev
+ARG VITE_WORKER_ENV=production
 ARG VITE_DAILY_API_KEY
 ARG VITE_RUNPOD_API_KEY
 ARG VITE_RUNPOD_IMAGE_ENDPOINT_ID
@@ -23,7 +23,8 @@ ARG VITE_RUNPOD_TEXT_ENDPOINT_ID
 ARG VITE_RUNPOD_WHISPER_ENDPOINT_ID
 
 # Set environment for build
-ENV VITE_TLDRAW_WORKER_URL=$VITE_TLDRAW_WORKER_URL
+# VITE_WORKER_ENV: 'production' | 'staging' | 'dev' | 'local'
+ENV VITE_WORKER_ENV=$VITE_WORKER_ENV
 ENV VITE_DAILY_API_KEY=$VITE_DAILY_API_KEY
 ENV VITE_RUNPOD_API_KEY=$VITE_RUNPOD_API_KEY
 ENV VITE_RUNPOD_IMAGE_ENDPOINT_ID=$VITE_RUNPOD_IMAGE_ENDPOINT_ID
diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml
index 43b1f86..e044009 100644
--- a/docker-compose.dev.yml
+++ b/docker-compose.dev.yml
@@ -8,7 +8,7 @@ services:
       context: .
       dockerfile: Dockerfile
       args:
-        - VITE_TLDRAW_WORKER_URL=https://jeffemmett-canvas.jeffemmett.workers.dev
+        - VITE_WORKER_ENV=staging
     container_name: canvas-dev
     restart: unless-stopped
     labels:
diff --git a/docker-compose.yml b/docker-compose.yml
index 478b048..314b6ba 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -8,7 +8,7 @@ services:
       context: .
       dockerfile: Dockerfile
       args:
-        - VITE_TLDRAW_WORKER_URL=https://jeffemmett-canvas.jeffemmett.workers.dev
+        - VITE_WORKER_ENV=production
         # Add other build args from .env if needed
     container_name: canvas-website
     restart: unless-stopped