diff --git a/debi.sh b/debi.sh
index 42b5434..a52edd7 100755
--- a/debi.sh
+++ b/debi.sh
@@ -214,7 +214,7 @@ sudo_with_password=false
 timezone=UTC
 ntp=time.google.com
 disk_partitioning=true
-disk=$(df /boot | grep -Eo '^/dev/[hsv]d[a-z]')
+disk="/dev/$(lsblk -no PKNAME "$(df /boot | grep -Eo '/dev/[a-z0-9]+')")"
 force_gpt=true
 efi=
 esp=106