From b827e81bc5f1f054085714e44149c0f6e6962906 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Hertzog?= <hertzog@debian.org>
Date: Wed, 4 Feb 2015 15:45:46 +0100
Subject: [PATCH] Fix logic to define the image extension

---
 build.sh | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/build.sh b/build.sh
index d38115c..f25d0db 100755
--- a/build.sh
+++ b/build.sh
@@ -138,7 +138,9 @@ for KALI_ARCH in $KALI_ARCHES; do
 	fi
 	set -e
 	IMAGE_EXT="${IMAGE_NAME##*.}"
-	IMAGE_EXT="${IMAGE_EXT:-img}"
+	if [ "$IMAGE_EXT" = "$IMAGE_NAME" ]; then
+		IMAGE_EXT="img"
+	fi
 	mv -f $IMAGE_NAME $TARGET_DIR/kali-linux-$KALI_VERSION-$KALI_ARCH.$IMAGE_EXT
 	mv -f build.log $TARGET_DIR/kali-linux-$KALI_VERSION-$KALI_ARCH.log
 done