diff --git a/src/install.sh b/src/install.sh
index ba22ad4..40b9fd2 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -412,7 +412,7 @@ updateImage() {
     return 1
   fi
 
-  if [[ "$2" == "win7"* ]]; then
+  if [[ "$3" == "win7"* ]]; then
 
     ETFS="boot.img"
     BOOT_MODE="windows_legacy"