diff --git a/scripts/3-post-setup.sh b/scripts/3-post-setup.sh
index f3e1857..d276a48 100755
--- a/scripts/3-post-setup.sh
+++ b/scripts/3-post-setup.sh
@@ -33,7 +33,7 @@ fi
 # set kernel parameter for adding splash screen
 sed -i 's/GRUB_CMDLINE_LINUX_DEFAULT="[^"]*/& splash /' /etc/default/grub
 
-if [ "$GRUB_THEME" == "cyberRE"]; then
+if [[ ${GRUB_THEME} == "cyberRE" ]]; then
   echo -e "Installing CyberRe Grub theme..."
   THEME_DIR="/boot/grub/themes"
   THEME_NAME=CyberRe
@@ -49,7 +49,7 @@ if [ "$GRUB_THEME" == "cyberRE"]; then
   echo "GRUB_THEME=\"${THEME_DIR}/${THEME_NAME}/theme.txt\"" >> /etc/default/grub
 fi
 
-if [ "$GRUB_INSTALL" == "yes"]; then
+if [[ ${GRUB_INSTALL} == "yes" ]]; then
 echo -e "Updating grub..."
 grub-mkconfig -o /boot/grub/grub.cfg
 echo -e "All set!"