diff --git a/Dockerfile b/Dockerfile
index a1fb642..994f0f7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,8 @@
+################################################################################
+## GO BUILDER
+################################################################################
 FROM golang:1.11.3 as builder
 
-EXPOSE 8080
 ENV VERSION 1.5.1
 ENV BUILD_DIR /build
 
@@ -14,7 +16,12 @@ COPY . .
 RUN go test -cover ./...
 RUN CGO_ENABLED=0 go build -a -tags netgo -installsuffix netgo -ldflags "-X github.com/halverneus/static-file-server/cli/version.version=${VERSION}" -o /serve /build/bin/serve
 
+################################################################################
+## DEPLOYMENT CONTAINER
+################################################################################
 FROM scratch
+
+EXPOSE 8080
 COPY --from=builder /serve /
 ENTRYPOINT ["/serve"]
 CMD []
diff --git a/Dockerfile.all b/Dockerfile.all
index 7adf054..b30b295 100644
--- a/Dockerfile.all
+++ b/Dockerfile.all
@@ -1,6 +1,5 @@
 FROM golang:1.11.3 as builder
 
-EXPOSE 8080
 ENV VERSION 1.5.1
 ENV BUILD_DIR /build