diff --git a/scripts/3-post-setup.sh b/scripts/3-post-setup.sh
index 0e94c72..f0e156f 100755
--- a/scripts/3-post-setup.sh
+++ b/scripts/3-post-setup.sh
@@ -18,8 +18,12 @@ GRUB EFI Bootloader Install & Check
 "
 source ${HOME}/ArchTitus/configs/setup.conf
 
+
 if [[ -d "/sys/firmware/efi" ]]; then
-    grub-install --efi-directory=/boot ${DISK}
+  if [[ $INSTALL_IN = "PART" ]]; then
+    grub-install --efi-directory=/boot/EFI ${PART}
+  elif [[ $INSTALL_IN = "DISK" ]]; then
+    grub-install --efi-directory=/boot/EFI ${DISK}
 fi
 
 echo -ne "