diff --git a/run/disk.sh b/run/disk.sh
index abdd2a8..f10bcbe 100644
--- a/run/disk.sh
+++ b/run/disk.sh
@@ -23,7 +23,7 @@ if [ -f "${DATA}" ]; then
 
     echo "INFO: Resizing data disk from $OLD_SIZE to $DATA_SIZE bytes.."
 
-    if [ "$ALLOCATE" = "N" ]; then
+    if [[ "${ALLOCATE}" == [Nn]* ]]; then
 
       # Resize file by changing its length
       truncate -s "${DATA_SIZE}" "${DATA}"; 
@@ -45,7 +45,7 @@ if [ -f "${DATA}" ]; then
         echo "ERROR: Could not allocate a file for the data disk." && exit 85
       fi
 
-      if [ "$ALLOCATE" = "Z" ]; then
+      if [[ "${ALLOCATE}" == [Zz]* ]]; then
 
         GB=$(( (REQ + 1073741823)/1073741824 ))
 
@@ -68,7 +68,7 @@ fi
 
 if [ ! -f "${DATA}" ]; then
 
-  if [ "$ALLOCATE" = "N" ]; then
+  if [[ "${ALLOCATE}" == [Nn]* ]]; then
 
     # Create an empty file
     truncate -s "${DATA_SIZE}" "${DATA}"
@@ -89,7 +89,7 @@ if [ ! -f "${DATA}" ]; then
       echo "ERROR: Could not allocate a file for the data disk." && exit 87
     fi
 
-    if [ "$ALLOCATE" = "Z" ]; then
+    if [[ "${ALLOCATE}" == [Zz]* ]]; then
 
       echo "INFO: Preallocating ${DISK_SIZE} of diskspace, please wait..."
       dd if=/dev/urandom of="${DATA}" count="${DATA_SIZE}" bs=1M iflag=count_bytes status=none