diff --git a/build.sh b/build.sh
index aeca79f..5c62b08 100755
--- a/build.sh
+++ b/build.sh
@@ -97,6 +97,7 @@ esac
 DOCKER_TAG="${DOCKER_TAG-${DOCKER_ORG}/${DOCKER_REPO}:${TAG}}"
 
 # Checking which VARIANT to build
+VARIANT="${VARIANT-main}"
 if [ "$VARIANT" == "main" ]; then
   DOCKERFILE="Dockerfile"
 else