From 23b756c1eb61dd369b6c002e0ed73f47c3d01986 Mon Sep 17 00:00:00 2001
From: Stuart Hayhurst <stuart.a.hayhurst@gmail.com>
Date: Sat, 17 Oct 2020 13:38:26 +0100
Subject: [PATCH] Fix --boot not saving through authentication

---
 install.sh | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/install.sh b/install.sh
index e52c3e9..41819e3 100755
--- a/install.sh
+++ b/install.sh
@@ -195,7 +195,11 @@ install() {
         read -p "[ Trusted ] Specify the root password : " -t${MAX_DELAY} -s
         [[ -n "$REPLY" ]] && {
         if [[ -n "${theme}" && -n "${screen}" ]]; then
+          if [[ ${THEME_DIR} == '/boot/grub/themes' ]]; then
+            sudo -S <<< $REPLY $0 --boot --${theme} --${icon} --${screen} --${custom_background}
+          else
             sudo -S <<< $REPLY $0 --${theme} --${icon} --${screen} --${custom_background}
+          fi
         fi
         } || {
              operation_canceled
@@ -342,7 +346,11 @@ remove() {
     read -p "[ trusted ] specify the root password : " -t${MAX_DELAY} -s
     [[ -n "$REPLY" ]] && {
       if [[ -n "${theme}" ]]; then
-        sudo -S <<< $REPLY $0 --remove --${theme}
+        if [[ ${THEME_DIR} == '/boot/grub/themes' ]]; then
+          sudo -S <<< $REPLY $0 --remove --boot --${theme}
+        else
+          sudo -S <<< $REPLY $0 --remove --${theme}
+        fi
       fi
     } || {
       operation_canceled