diff --git a/partition.sh b/partition.sh
index 7dba356..9afa63e 100755
--- a/partition.sh
+++ b/partition.sh
@@ -62,7 +62,9 @@ if [[ "${1}" == '--disk' ]]; then
 		-t3:BF00 \
 		-c3:${ID} \
 		${disk}
-elif [[ "${1}" == '--efi' ]]; then
+fi
+
+if [[ "${1}" == '--efi' ]]; then
 	if [[ ! "${disk}" == **/dev/disk/by-id/** ]]; then
 		if [[ "${disk}" == **/dev/nvme** ]]; then
 			part1='p1'