diff --git a/installer.sh b/installer.sh
index 876a878..baccdca 100755
--- a/installer.sh
+++ b/installer.sh
@@ -47,6 +47,17 @@ insmod /root/vfat-modules/vfat.ko.zst
 dialog --title "Welcome" --msgbox "Welcome to the EvolutionOS installer! \n\nThis installer is meant to be straightforward, no need for technical skill. \n\nPress enter to select buttons, tab to move between text boxes, and left / right to move between buttons." 0 0
 choice=$(dialog --menu "Select an installation drive:" 20 60 10 $(create_menu $(parted -l | grep -oE '^Disk /dev/[a-zA-Z0-9]+' | cut -d' ' -f2- | cut -c 6-)) 2>&1 >/dev/tty)
 
+if [[ ${choice} =~ "nvme" ]]; then
+    part1="p1"
+    part2="p2"
+elif [[ ${choice} =~ "mmcblk" ]]; then
+    part1="p1"
+    part2="p2"
+else
+    part1="1"
+    part2="2"
+fi
+
 if [[ "$(cat /sys/block/${choice}/ro)" = "1" ]]; then
     dialog --title "Scan disk" --msgbox "You are not able to install evolutionOS on a non-writable disk" 0 0
     failinstall