diff --git a/storage-test.sh b/storage-test.sh
index 0d66828..96030b3 100755
--- a/storage-test.sh
+++ b/storage-test.sh
@@ -104,13 +104,13 @@ IO4kRandWrite=0
 # Did the user give a folder ?
 ChosenPartition=""
 if [ "$1" = "" ]; then
-  # User did not provide a partition/folder, change directory to rootfs
-  df=0
+  df=0; declare -A driveSel
   drives="$(/bin/df -kTh | grep ext4)"
-  echo "$drives"
   echo "$drives" | while IFS= read -r line ; do
 	  ((df++))
-	  echo "$df) $line"
+	  path=$(echo $line | cut -d'%' -f2)
+	  ${driveSel[${df}]}=${path// }
+	  echo "$df) ${path// }"
   done
   
   exit 0