diff --git a/src/install.sh b/src/install.sh
index b0d58d9..084b33e 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -838,13 +838,15 @@ copyOEM() {
 
 buildImage() {
 
-  local dir="$1"
+  local iso="$1"
+  local dir="$2"
   local failed="N"
   local cat="BOOT.CAT"
   local log="/run/shm/iso.log"
-  local size size_gb space space_gb desc
-  local out="$TMP/${BASE%.*}.tmp"
+  local base size size_gb space space_gb desc
 
+  base=$(basename "$iso")
+  local out="$TMP/${base%.*}.tmp"
   rm -f "$out"
 
   desc=$(printVersion "$DETECTED" "ISO")
@@ -1007,7 +1009,7 @@ if ! copyOEM "$DIR"; then
   exit 63
 fi
 
-if ! buildImage "$DIR"; then
+if ! buildImage "$ISO" "$DIR"; then
   exit 65
 fi