diff --git a/run/install.sh b/run/install.sh
index b19208b..5d3a3d6 100644
--- a/run/install.sh
+++ b/run/install.sh
@@ -125,17 +125,24 @@ if [ "$ALLOCATE" != "F" ]; then
 else
 
   MB=$(( (SYSTEM_SIZE + 1048575)/1048576 ))
+  echo "INFO: Writing ${MB} MB of zeroes, please wait.."
 
-  echo "INFO: Writing ${MB} MB of zeroes.."
   dd if=/dev/zero of="${SYSTEM}" count="${MB}" bs=1M
   truncate -s "${SYSTEM_SIZE}" "${SYSTEM}"
 
 fi
 
+# Check if file exists
+if [ ! -f "${SYSTEM}" ]; then
+    echo "ERROR: System disk does not exist ($SYSTEM)" && exit 89
+fi
+
+# Check the filesize
 SIZE=$(stat -c%s "${SYSTEM}")
 
 if [[ SIZE -ne SYSTEM_SIZE ]]; then
-  echo "ERROR: System disk has the wrong size." && exit 89
+  rm -f "${SYSTEM}"
+  echo "ERROR: System disk has the wrong size: ${SIZE}" && exit 90
 fi
 
 PART="$TMP/partition.fdisk"