diff --git a/Dockerfile b/Dockerfile
index 3c39471..e1452b5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,10 +2,11 @@ FROM golang:1.26-alpine AS build
 WORKDIR /src
 COPY go.* ./
 RUN go mod download
-COPY cmd ./cmd
+RUN go mod tidy
+COPY . .
 COPY internal ./internal
-RUN CGO_ENABLED=0 GOOS=linux go build -trimpath -ldflags="-s -w" -o /out/neuralhunt ./cmd/server \
- && CGO_ENABLED=0 GOOS=linux go build -trimpath -ldflags="-s -w" -o /out/neuralhunt-client ./cmd/client
+RUN CGO_ENABLED=0 GOOS=linux go build -trimpath -ldflags="-s -w" -o /out/neuralhunt ./cmd/server
+RUN CGO_ENABLED=0 GOOS=linux go build -trimpath -ldflags="-s -w" -o /out/neuralhunt-client ./cmd/client
 
 FROM alpine:3.24
 RUN adduser -D -H app && mkdir -p /data /app && chown -R app:app /data /app