diff --git a/install.bash b/install.bash
index bf97bda..67add9f 100644
--- a/install.bash
+++ b/install.bash
@@ -6,8 +6,15 @@ install() {
 	local encrypt_disk="${2}"
 	local host="${3}"
 
-	local boot_partition="${disk}1"
-	local primary_partition="${disk}2"
+	local boot_partition=""
+	local primary_partition=""
+	if [[ "${disk}" == *"nvme"* ]]; then
+		local boot_partition="${disk}p1"
+		local primary_partition="${disk}p2"
+	else
+		local boot_partition="${disk}1"
+		local primary_partition="${disk}2"
+	fi
 	# The size is large because I'd like to be able to hibernate my laptop in its entirety. I have 64 GB of ram.
 
 	local label_crypt_luks="NixOS-Crypt"