diff --git a/Dockerfile b/Dockerfile
index 575d574..e142877 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,11 @@
 # ------------------------------------------------------------------------
-from mcluseau/golang-builder:1.20.4 as build
+from golang:1.21.3-bullseye as build
+workdir /src
+copy go.mod go.sum ./
+run go mod download
+copy . ./
+run go test ./...
+run hack/build ./...
 
 # ------------------------------------------------------------------------
 from debian:bullseye
@@ -7,13 +13,8 @@ entrypoint ["/bin/dkl-local-server"]
 
 env _uncache 1
 run apt-get update \
- && apt-get install -y genisoimage gdisk dosfstools util-linux udev binutils systemd \
+ && yes |apt-get install -y genisoimage gdisk dosfstools util-linux udev binutils systemd \
+ grub2 grub-pc-bin grub-efi-amd64-bin ca-certificates curl openssh-client \
  && apt-get clean
 
-run yes |apt-get install -y grub2 grub-pc-bin grub-efi-amd64-bin \
- && apt-get clean
-
-run apt-get install -y ca-certificates curl openssh-client \
- && apt-get clean
-
-copy --from=build /go/bin/ /bin/
+copy --from=build /src/dist/ /bin/