From a2318c2e49df2a461cccd2fdd4079353b6c7fa58 Mon Sep 17 00:00:00 2001
From: Amelia Carter <lideemily@gmail.com>
Date: Sat, 14 Nov 2015 21:01:30 -0800
Subject: [PATCH] Added test for wm variable

---
 ufetch-arch   | 4 +++-
 ufetch-gentoo | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

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