diff --git a/neofetch b/neofetch
index 3753d57b..ba84496e 100755
--- a/neofetch
+++ b/neofetch
@@ -2192,8 +2192,8 @@ get_de() {
     # TODO:
     #  - New config option + flag: --de_display_server on/off ?
     #  - Add display of X11, Arcan and anything else relevant.
-    [[ $de && $WAYLAND_DISPLAY ]] &&
-        de+=" (Wayland)"
+    [[ $de ]] &&
+        de+=" (${XDG_SESSION_TYPE})"
 
     de_run=1
 }