diff --git a/disk.sh b/disk.sh
index f8d42ed..997e10d 100644
--- a/disk.sh
+++ b/disk.sh
@@ -14,7 +14,10 @@ DATA="$IMG/data$DISK_SIZE.img"
 
 if [ ! -f "$DATA" ]; then
     # Create an empty file
-    fallocate -l "${NEW_SIZE}" "${DATA}"
+    if ! fallocate -l "${NEW_SIZE}" "${DATA}"; then
+      rm -f "${DATA}"
+      echo "ERROR: Not enough free space to create virtual disk." && exit 88
+    fi
     # Format as BTRFS filesystem
     mkfs.btrfs -q -L data -d single -m dup "${DATA}" > /dev/null
 fi
diff --git a/install.sh b/install.sh
index 9aeec4c..0f0101a 100644
--- a/install.sh
+++ b/install.sh
@@ -101,7 +101,11 @@ echo "Install: Creating partition table..."
 
 SYSTEM="$TMP/sys.img"
 SYSTEM_SIZE="4954537983"
-fallocate -l "${SYSTEM_SIZE}" "${SYSTEM}"
+
+if ! fallocate -l "${SYSTEM_SIZE}" "${SYSTEM}"; then
+  rm -f "${SYSTEM}"
+  echo "ERROR: Not enough free space to create virtual disk." && exit 88
+fi
 
 PART="$TMP/partition.fdisk"