diff --git a/controller/Dockerfile b/controller/Dockerfile
index 448b256..9bac456 100644
--- a/controller/Dockerfile
+++ b/controller/Dockerfile
@@ -15,12 +15,15 @@ ARG GIT_COMMIT=unknown
 WORKDIR /build
 
 # Cache dependencies first
-COPY go.mod go.sum ./
-RUN go mod download
+COPY go.mod ./
+RUN go mod download || true
 
 # Copy source
 COPY . .
 
+# Generate go.sum and ensure all deps are fetched
+RUN go mod tidy
+
 # Build static binary
 RUN CGO_ENABLED=0 GOOS=${TARGETOS} GOARCH=${TARGETARCH} go build \
     -ldflags="-s -w \
@@ -78,4 +81,4 @@ HEALTHCHECK --interval=30s --timeout=5s --start-period=10s --retries=3 \
 EXPOSE 8080
 
 ENTRYPOINT ["/usr/local/bin/felhom-controller"]
-CMD ["--config", "/opt/docker/felhom-controller/controller.yaml"]
+CMD ["--config", "/opt/docker/felhom-controller/controller.yaml"]
\ No newline at end of file