diff --git a/nixos/maintainers/scripts/azure-new/upload-image.sh b/nixos/maintainers/scripts/azure-new/upload-image.sh
index 4dd5cdfd019f..1466dcd1f0a2 100755
--- a/nixos/maintainers/scripts/azure-new/upload-image.sh
+++ b/nixos/maintainers/scripts/azure-new/upload-image.sh
@@ -2,9 +2,9 @@
 set -euo pipefail
 set -x
 
-attr="${1:-"azbasic"}"
+image_nix="${1:-"./examples/basic/image.nix"}"
 
-nix-build ./examples/basic/image.nix --out-link "azure"
+nix-build "${image_nix}" --out-link "azure"
 
 group="nixos-images"
 location="westus2"