diff --git a/app.dockerfile b/app.dockerfile
index 6878528..8c425f3 100644
--- a/app.dockerfile
+++ b/app.dockerfile
@@ -21,6 +21,9 @@ RUN yarn install --frozen-lockfile && yarn build
 
 FROM node:22-alpine
 
+ENV NEXT_PUBLIC_WS_URL=ws://localhost:3001
+ENV NEXT_PUBLIC_API_URL=http://localhost:3001/api
+
 WORKDIR /app
 
 # Copy built assets from the builder stage
diff --git a/backend.dockerfile b/backend.dockerfile
index 9ac1c4c..deed1de 100644
--- a/backend.dockerfile
+++ b/backend.dockerfile
@@ -29,9 +29,9 @@ COPY --chown=node:node --from=builder /app/dist ./dist
 COPY --chown=node:node --from=builder /app/node_modules ./node_modules
 
 # Copy the rest of the application code
+COPY --chown=node:node src ./src
 COPY --chown=node:node drizzle.config.ts ./
 COPY --chown=node:node tsconfig.json ./
-COPY --chown=node:node src/db/schema.ts ./src/db/schema.ts
 COPY --chown=node:node package.json ./package.json
 
 # Create data directory & set permissions to node user