diff --git a/nix/docker/heptapod/image.json b/nix/docker/heptapod/image.json
index 61cd890913..7effeefd1a 100644
--- a/nix/docker/heptapod/image.json
+++ b/nix/docker/heptapod/image.json
@@ -1,7 +1,7 @@
 {
   "imageName": "octobus/heptapod",
-  "imageDigest": "sha256:af6a7f47a15410c521a0d620377b98fa6f5715d6f091ea39d7e332146d20786c",
-  "sha256": "1gdi9q02g2a5y2vmpxray4l8rq3yapqpdbg0fg7xxk9f99ysng7j",
+  "imageDigest": "sha256:3f854f6c21a4a66c8decb9e05f9e31b4a29c49acdd5c4b2539861a02610088bf",
+  "sha256": "sha256:0xm81pa48z0h6a4dm10ih21i08c63q23qw3610s3kiznj19m1hah",
   "finalImageName": "octobus/heptapod",
-  "finalImageTag": "0.30.1"
+  "finalImageTag": "0.31.2"
 }
diff --git a/nix/docker/heptapod/update.sh b/nix/docker/heptapod/update.sh
index 2d8dd64cc5..e797266e5b 100755
--- a/nix/docker/heptapod/update.sh
+++ b/nix/docker/heptapod/update.sh
@@ -7,7 +7,12 @@ repo="index.docker.io/octobus/heptapod"
 path="nix.docker.heptapod"
 depot="$(pwd | grep -o '.*/depot')"
 
-latest_digest="$(crane digest "$repo:latest")"
+want_tag="latest"
+if [[ ! -z "$1" ]]; then
+  want_tag="$1"
+fi
+
+latest_digest="$(crane digest "$repo:$want_tag")"
 current_digest="$(jq -r .imageDigest image.json)"
 
 if [[ "$latest_digest" == "$current_digest" ]]; then