2 files changed, 4 insertions, 1 deletions
diff --git a/install.sh b/install.sh
index adaa8bd..557e905 100755
--- a/install.sh
+++ b/install.sh
@@ -56,6 +56,9 @@ else
   exit 1
 fi
 
+_numBoot="${partition_array[0]}"
+export _numBoot
+
 export disk_drive
 export root_drive
 export boot_drive
diff --git a/post_chroot.sh b/post_chroot.sh
index db0d456..c61e81d 100755
--- a/post_chroot.sh
+++ b/post_chroot.sh
@@ -46,7 +46,7 @@ elif [ "$_kernelflag" -eq 2 ]; then
   mkdir -p /boot/EFI/BOOT
   cp arch/x86/boot/bzImage /boot/EFI/BOOT/BOOTX64.EFI
   _diskdrivewop="${diskdrive%p}"
-  efibootmgr -c -d /dev/$_diskdrivewop -p ${partition_array[0]} -L "linux" -l '\EFI\BOOT\BOOTX64.EFI'
+  efibootmgr -c -d /dev/$_diskdrivewop -p $_numBoot -L "linux" -l '\EFI\BOOT\BOOTX64.EFI'
 else
   printf ${LIGHTRED}"Wrong kernelflag value.${NoColor}\n"
   exit 1