diff --git a/install.sh b/install.sh
index c654099..17ca7e8 100644
--- a/install.sh
+++ b/install.sh
@@ -164,7 +164,12 @@ lsblk | grep "$dev"
 echo
 
 # Get new variable.
-main=${dev}2
+if [ $dev = "nvme0n1" ];
+  then
+    main=${dev}p2
+  else
+    main=${dev}2
+fi
 
 # Encrypting partition.
 echo "Encrypting system partition. This might take a while."
@@ -184,7 +189,14 @@ lsblk | grep "$dev"
 echo
 
 echo "Creating the filesystem."
-boot=${dev}1
+
+if [ $dev = "nvme0n1" ];
+  then
+    boot=${dev}p2
+  else
+    boot=${dev}1
+fi
+
 mkfs.fat -F 32 -n UEFI /dev/$boot &&
 echo "Filesystem for boot successfully created."
 echo