diff --git a/Dockerfile b/Dockerfile
index 4cd82699..77b88b9c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -35,6 +35,10 @@ RUN apt-get update && apt-get install -y make
 WORKDIR /app
 RUN make build-web
 
+# Copy the frontend build from the frontend stage
+# COPY --from=frontend /app/web /checkpointz
+RUN serve -s build -l 5555
+
 # Stage 3: Create the final image
 FROM ubuntu:latest
 
@@ -48,10 +52,6 @@ RUN apt-get update && apt-get -y upgrade && apt-get install -y --no-install-reco
 # Copy the Go binary from the builder stage
 COPY --from=builder /bin/app /checkpointz
 
-# Copy the frontend build from the frontend stage
-# COPY --from=frontend /app/web /checkpointz
-RUN serve -s build -l 5555
-
 # Expose port 5555
 EXPOSE 5555