diff --git a/modules/shell/zsh.nix b/modules/shell/zsh.nix
index cc09303..4b3ad2e 100644
--- a/modules/shell/zsh.nix
+++ b/modules/shell/zsh.nix
@@ -19,7 +19,7 @@
       };
 
       initContent = ''
-        if [[ $DISPLAY ]]; then
+        if [[ $WAYLAND_DISPLAY ]]; then
             [[ $- != *i* ]] && return
             [[ -z "$TMUX" ]] && (tmux attach || tmux new-session)
         fi