diff --git a/readme.md b/readme.md
index 0865037b..d065f598 100644
--- a/readme.md
+++ b/readme.md
@@ -90,7 +90,7 @@ docker run -it --rm --name qemu -e "BOOT=http://example.com/image.iso" -p 8006:8
   
   ```yaml
   volumes:
-    - /home/user/example.iso:/storage/boot.iso
+    - /home/user/example.iso:/boot.iso
   ```
 
   Replace the example path `/home/user/example.iso` with the filename of the desired ISO file.
diff --git a/src/disk.sh b/src/disk.sh
index d34c96d9..b32abec8 100644
--- a/src/disk.sh
+++ b/src/disk.sh
@@ -10,8 +10,7 @@ set -Eeuo pipefail
 : "${DISK_DISCARD:="on"}"         # Controls whether unmap (TRIM) commands are passed to the host.
 : "${DISK_ROTATION:="1"}"         # Rotation rate, set to 1 for SSD storage and increase for HDD
 
-BOOT="$STORAGE/$BASE"
-[ ! -f "$BOOT" ] && BOOT="/dev/null"
+[ ! -f "$BOOT" ] || [ ! -s "$BOOT" ] && BOOT="/dev/null"
 
 DISK_OPTS="-object iothread,id=io2"
 DISK_OPTS="$DISK_OPTS -drive id=cdrom0,media=cdrom,if=none,format=raw,readonly=on,file=$BOOT"
@@ -24,7 +23,7 @@ else
 fi
 
 DRIVERS="$STORAGE/drivers.iso"
-[ ! -f "$DRIVERS" ] && DRIVERS="/run/drivers.iso"
+[ ! -f "$DRIVERS" ] || [ ! -s "$DRIVERS" ] && DRIVERS="/run/drivers.iso"
 
 if [ -f "$DRIVERS" ] && [[ "${MACHINE,,}" != "pc-q35-2"* ]]; then
   DISK_OPTS="$DISK_OPTS -drive id=cdrom1,media=cdrom,if=none,format=raw,readonly=on,file=$DRIVERS"
diff --git a/src/install.sh b/src/install.sh
index ae36972d..098881ab 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -3,36 +3,41 @@ set -Eeuo pipefail
 
 # Check if running with interactive TTY or redirected to docker log
 if [ -t 1 ]; then
-  PROGRESS="--progress=bar:noscroll"
+  progress="--progress=bar:noscroll"
 else
-  PROGRESS="--progress=dot:giga"
+  progress="--progress=dot:giga"
 fi
 
-BASE=$(find "$STORAGE" -maxdepth 1 -type f -iname boot.iso -printf "%f\n" | head -n 1)
-[ -z "$BASE" ] && BASE=$(find "$STORAGE" -maxdepth 1 -type f -iname boot.img -printf "%f\n" | head -n 1)
+file="/boot.iso" && [ -f "$file" ] && [ -s "$file" ] && BOOT="$file" && return 0
+file="/boot.img" && [ -f "$file" ] && [ -s "$file" ] && BOOT="$file" && return 0
 
-[ -f "$STORAGE/$BASE" ] && return 0
+file=$(find "$STORAGE" -maxdepth 1 -type f -iname boot.iso -printf "%f\n" | head -n 1)
+[ -z "$file" ] && file=$(find "$STORAGE" -maxdepth 1 -type f -iname boot.img -printf "%f\n" | head -n 1)
+[ -n "$file" ] && file="$STORAGE/$file" 
+[ -f "$file" ] && [ -s "$file" ] && BOOT="$file" && return 0
 
 if [ -z "$BOOT" ]; then
   error "No boot disk specified, set BOOT= to the URL of an ISO file." && exit 64
 fi
 
-BASE=$(basename "$BOOT")
-[ -f "$STORAGE/$BASE" ] && return 0
+base=$(basename "$BOOT")
+[ -n "$base" ] && file="$STORAGE/$base" 
+[ -f "$file" ] && [ -s "$file" ] && BOOT="$file" && return 0
 
-BASE=$(basename "${BOOT%%\?*}")
-: "${BASE//+/ }"; printf -v BASE '%b' "${_//%/\\x}"
-BASE=$(echo "$BASE" | sed -e 's/[^A-Za-z0-9._-]/_/g')
-[ -f "$STORAGE/$BASE" ] && return 0
+base=$(basename "${BOOT%%\?*}")
+: "${base//+/ }"; printf -v base '%b' "${_//%/\\x}"
+base=$(echo "$base" | sed -e 's/[^A-Za-z0-9._-]/_/g')
+[ -n "$base" ] && file="$STORAGE/$base" 
+[ -f "$file" ] && [ -s "$file" ] && BOOT="$file" && return 0
 
-TMP="$STORAGE/${BASE%.*}.tmp"
+TMP="$STORAGE/${base%.*}.tmp"
 rm -f "$TMP"
 
-MSG="Downloading $BASE..."
-info "$MSG" && html "$MSG"
+msg="Downloading $base..."
+info "$msg" && html "$msg"
 
-/run/progress.sh "$TMP" "" "Downloading $BASE ([P])..." &
-{ wget "$BOOT" -O "$TMP" -q --timeout=10 --show-progress "$PROGRESS"; rc=$?; } || :
+/run/progress.sh "$TMP" "" "Downloading $base ([P])..." &
+{ wget "$BOOT" -O "$TMP" -q --timeout=10 --show-progress "$progress"; rc=$?; } || :
 
 fKill "progress.sh"
 
@@ -41,12 +46,12 @@ fKill "progress.sh"
 
 html "Download finished successfully..."
 
-SIZE=$(stat -c%s "$TMP")
+size=$(stat -c%s "$TMP")
 
-if ((SIZE<100000)); then
+if ((size<100000)); then
   error "Invalid ISO file: Size is smaller than 100 KB" && exit 62
 fi
 
-mv -f "$TMP" "$STORAGE/$BASE"
+mv -f "$TMP" "$file"
 
 return 0