diff --git a/storage-test.sh b/storage-test.sh
index 4cb8da1..2ae6c05 100755
--- a/storage-test.sh
+++ b/storage-test.sh
@@ -331,7 +331,7 @@ fi
 # Run sync to make sure all changes have been written to disk
 sync
 
-if [ -z "$ChosenPartition" ]; then
+if [ "$ChosenPartition" != "" ]; then
   # User did not provide a partition/folder, continue with rootfs
   # --Get system boot drive information--
   # Find from mountpoint first
@@ -380,7 +380,7 @@ if [ -z "$BootDriveSuffix" ]; then
   BootDriveSuffix="$BootDrive"
 fi
 
-if [ -z "$ChosenPartition" ]; then
+if [ "$ChosenPartition" != "" ]; then
   Print_Style "System rootfs drive (/) has been detected as $BootDrive ($BootDriveSuffix)" "$YELLOW"
 else
   Print_Style "Chosen partition ($ChosenPartition) has been detected as $BootDrive ($BootDriveSuffix)" "$YELLOW"