diff --git a/bootstrap.sh b/bootstrap.sh
index d2e9a94..9ec714c 100755
--- a/bootstrap.sh
+++ b/bootstrap.sh
@@ -17,7 +17,7 @@ init() {
         read -sp "Repeat: " CHECK
     done
     # Show some possible disks
-    [ -z "$DEV" ] && lsblk -nrpo "name,size,model" && read -p "Provide installation medium (e.g. /dev/sda): " DEV
+    [ -z "$DEV" ] && lsblk -nrpo "name,size,model" && read -p "Provide installation medium (e.g. sda, nvme0n1): " DEV
     [[ "$DEV" =~ sd[a-z] ]] && SUF="1-3" && MODE="SATA"
     [[ "$DEV" =~ nvme[0-9]n[0-9] ]] && SUF="p1-3" && MODE="NVME"
 
@@ -39,7 +39,7 @@ set -e
 init
 
 # All values set, start bootstrapping
-gdisk <<EOF
+gdisk /dev/$DEV <<EOF
 o
 y
 n