diff --git a/ufetch-arch b/ufetch-arch
index a806e19..7064503 100755
--- a/ufetch-arch
+++ b/ufetch-arch
@@ -12,7 +12,9 @@ KERNEL="$(uname -sr)"
 UPTIME="$(uptime -p | sed s:"up "::)"
 PACKAGES="$(pacman -Q | wc -l)"
 # shell is already defined
-WM="$(grep -v "^#" "${HOME}/.xinitrc" | tail -n 1 | cut -d " " -f2)"
+if [[ -z "$WM" ]]; then
+	WM="$(grep -v "^#" "${HOME}/.xinitrc" | tail -n 1 | cut -d " " -f2)"
+fi
 
 ## DEFINE COLORS
 
diff --git a/ufetch-gentoo b/ufetch-gentoo
index da3ae98..f26e34b 100755
--- a/ufetch-gentoo
+++ b/ufetch-gentoo
@@ -12,7 +12,9 @@ KERNEL="$(uname -sr)"
 UPTIME="$(uptime -p | sed s:'up ':'':)"
 PACKAGES="$(ls -d /var/db/pkg/*/* | wc -l)"
 # shell is already defined
-WM=$(tail -n 1 "$HOME/.xinitrc" | cut -d ' ' -f3)
+if [[ -z "$WM" ]]; then
+	WM=$(tail -n 1 "$HOME/.xinitrc" | cut -d ' ' -f3)
+fi
 
 ## DEFINE COLORS