diff --git a/build.sh b/build.sh
index 05363d0..5caf1a7 100644
--- a/build.sh
+++ b/build.sh
@@ -48,8 +48,8 @@ usage() {
 # --- parse arguments -------------------------------------------------------
 while [[ $# -gt 0 ]]; do
   case $1 in
-    -s|--snapshot)  VERSION="${IMAGE_TAG}-SNAPSHOT$(git rev-parse --short HEAD 2>/dev/null || echo "local")"; shift ;;
-    -r|--release)   VERSION="${IMAGE_TAG}"; shift ;;
+    -s|--snapshot)  VERSION="${IMAGE_VERSION}-SNAPSHOT$(git rev-parse --short HEAD 2>/dev/null || echo "local")"; shift ;;
+    -r|--release)   VERSION="${IMAGE_VERSION}"; shift ;;
     -v|--version)   VERSION="$2"; shift 2 ;;
     -p|--push)      PUSH=true; shift ;;
     -o|--only)      ONLY+=("$2"); shift 2 ;;