diff --git a/test.uefi b/test.uefi
index fdd0d51..37241e6 100644
--- a/test.uefi
+++ b/test.uefi
@@ -25,7 +25,7 @@ fi
 RECREATE_PARTITIONS="1"
 declare -A PARTITIONS_PRESENT=()
 
-CHECK_PARTITIONS_OUTPUT=$(lsblk -o NAME,PARTTYPE | grep "${DRIVE//\/dev\/}[0-9]")
+CHECK_PARTITIONS_OUTPUT=$(lsblk -po NAME,PARTTYPE | grep "${DRIVE}[0-9]")
 
 CHECK_ESP=$(echo "${CHECK_PARTITIONS_OUTPUT}" | grep -c "c12a7328-f81f-11d2-ba4b-00a0c93ec93b")
 PARTITIONS_PRESENT+=([ESP]="${CHECK_ESP}")