diff --git a/build.ps1 b/build.ps1
index 3c4da33..0303c5f 100755
--- a/build.ps1
+++ b/build.ps1
@@ -1,10 +1,10 @@
 # This script will run docker-compose-generator in a container to generate the yml files
 
 If (-not ($BTCPAYGEN_DOCKER_IMAGE)) { $BTCPAYGEN_DOCKER_IMAGE = "btcpayserver/docker-compose-generator" }
-If($BTCPAYGEN_DOCKER_IMAGE -eq "build"){
-	docker build docker-compose-generator --tag build
+If($BTCPAYGEN_DOCKER_IMAGE -eq "btcpayserver/docker-compose-generator:local"){
+	docker build docker-compose-generator --tag $BTCPAYGEN_DOCKER_IMAGE
 } Else {
-	docker pull $BTCPAYGEN_DOCKER_IMAGE:
+	docker pull $BTCPAYGEN_DOCKER_IMAGE
 }
 
 docker run -v "$(Get-Location)\Generated:/app/Generated" `
diff --git a/build.sh b/build.sh
index 2820aaa..b0d1b24 100755
--- a/build.sh
+++ b/build.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
 
-if [ "$BTCPAYGEN_DOCKER_IMAGE" == "build" ]
+if [ "$BTCPAYGEN_DOCKER_IMAGE" == "btcpayserver/docker-compose-generator:local" ]
 then
-    docker build docker-compose-generator --tag build
+    docker build docker-compose-generator --tag $BTCPAYGEN_DOCKER_IMAGE
 else
     docker pull ${BTCPAYGEN_DOCKER_IMAGE:-"btcpayserver/docker-compose-generator"}
 fi