diff --git a/Dockerfile b/Dockerfile
index 420047c..cab568b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,5 @@
-FROM ghcr.io/goauthentik/server:latest
+ARG TAG="latest"
+FROM ghcr.io/goauthentik/server:${TAG}
 
 COPY ./custom.css /web/dist/custom.css
 COPY ./assets/icons/ /web/dist/assets/icons/
diff --git a/build b/build
index b1d4017..e887b22 100755
--- a/build
+++ b/build
@@ -4,5 +4,5 @@ tag="${1:-latest}"
 image="$name:$tag"
 
 echo "Building $image..."
-podman build -t "$image" --pull .
+podman build -t "$image" --pull --build-arg "TAG=$tag" .
 podman push "$image"