diff --git a/Dockerfile b/Dockerfile
index 36ad71c..eaae1e6 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,11 +19,12 @@ COPY deno.json deno.lock* deps.ts ./
 COPY server/deno.json ./server/
 COPY sdk/deno.json ./sdk/
 COPY ui/deno.json ./ui/
+COPY src/deno.json ./src/
 RUN deno cache deps.ts
 
 # Source code layer
 COPY . .
-RUN deno cache server/main.ts
+RUN deno cache src/main.ts
 
 # Stage 3: Runner
 FROM denoland/deno:debian-2.9.4
@@ -42,4 +43,5 @@ COPY --from=deno-builder --chown=deno:deno /app ./
 EXPOSE 8000
 
 # Added --allow-ffi for Deno.dlopen and --allow-read for dlopen path resolution
-CMD ["run", "--allow-net", "--allow-env", "--allow-ffi=./libspire_ffi.so", "--allow-read=.,/var/run/spire/agent.sock", "--unstable-ffi", "server/main.ts"]
+CMD ["run", "--allow-net", "--allow-env", "--allow-ffi=./libspire_ffi.so", "--allow-read=.,/var/run/spire/agent.sock", "--unstable-ffi", "src/main.ts"]
+