From 2deb2bf03f452f824645abb55deb3252014cf0f1 Mon Sep 17 00:00:00 2001
From: WaLLy3K <WaLLy3K@users.noreply.github.com>
Date: Fri, 22 Sep 2017 02:23:43 +1000
Subject: [PATCH] Fix broken whitelist functionality

---
 gravity.sh | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/gravity.sh b/gravity.sh
index cb717aba..881fc6f2 100755
--- a/gravity.sh
+++ b/gravity.sh
@@ -631,8 +631,8 @@ fi
 
 gravity_ShowBlockCount
 
-# Perform when downloading blocklists, or modifying the blacklist
-if [[ "${skipDownload}" == false ]] || [[ "${listType}" == "blacklist" ]]; then
+# Perform when downloading blocklists, or modifying the white/blacklist (not wildcards)
+if [[ "${skipDownload}" == false ]] || [[ "${listType}" == *"list" ]]; then
   str="Parsing domains into hosts format"
   echo -ne "  ${INFO} ${str}..."
 
@@ -645,10 +645,7 @@ if [[ "${skipDownload}" == false ]] || [[ "${listType}" == "blacklist" ]]; then
   fi
 
   echo -e "${OVER}  ${TICK} ${str}"
-fi
 
-# Perform when downloading blocklists
-if [[ "${skipDownload}" == false ]]; then
   gravity_Cleanup
 fi