From 70152af6d26e7618bd8e9f86208eaf4ca9bf7311 Mon Sep 17 00:00:00 2001
From: Vince <vinceliuice@hotmail.com>
Date: Sun, 18 Oct 2020 22:24:56 +0800
Subject: [PATCH] Update install.sh

---
 install.sh | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/install.sh b/install.sh
index 591f360..03eed25 100755
--- a/install.sh
+++ b/install.sh
@@ -198,6 +198,7 @@ install() {
     prompt -i "\n Updating grub config...\n"
 
     updating_grub
+    prompt -w "\n * At the next restart of your computer you will see your new Grub theme: '$theme' "
   else
     # Error message
     prompt -e "\n [ Error! ] -> Run me as root! "
@@ -205,13 +206,13 @@ install() {
     # persisted execution of the script as root
     if [[ -n ${tui_root_login} ]] ; then
         if [[ -n "${theme}" && -n "${screen}" ]]; then
-            sudo -S <<< ${tui_root_login} $0 "${PROG_ARGS[@]}"
+            sudo -S <<< ${tui_root_login} $0 --${theme} --${icon} --${screen}
         fi
     else
         read -p "[ Trusted ] Specify the root password : " -t${MAX_DELAY} -s
         [[ -n "$REPLY" ]] && {
         if [[ -n "${theme}" && -n "${screen}" ]]; then
-          sudo -S <<< $REPLY "$0" "${PROG_ARGS[@]}"
+          sudo -S <<< $REPLY "$0" --${theme} --${icon} --${screen}
         fi
         } || {
              operation_canceled
@@ -294,7 +295,6 @@ updating_grub() {
 
   # Success message
   prompt -s "\n * All done!"
-  prompt -w "\n * At the next restart of your computer you will see your new Grub theme: '$theme' "
 }
 
 install_dialog() {