diff --git a/ufetch b/ufetch
index 97fdfbf..d80b404 100755
--- a/ufetch
+++ b/ufetch
@@ -12,7 +12,9 @@ KERNEL="$(uname -sr)"
 UPTIME="$(uptime -p | sed s:"up "::)"
 PACKAGES="$(pkginfo -i | wc -l)"
 # shell is already defined
-WM="$(tail -n 1 "${HOME}/.xinitrc" | cut -d " " -f2)"
+if [[ -z "$WM" ]]; then
+    WM="$(grep -v "^#" "${HOME}/.xinitrc" | tail -n 1 | cut -d " " -f2)"
+fi
 
 ## DEFINE COLORS