diff --git a/resources/niri-session b/resources/niri-session
index ff9a5c6f..49dc4c94 100755
--- a/resources/niri-session
+++ b/resources/niri-session
@@ -50,7 +50,7 @@ if hash systemctl >/dev/null 2>&1; then
     systemctl --user start --job-mode=replace-irreversibly niri-shutdown.target
 
     # Unset environment that we've set.
-    systemctl --user unset-environment WAYLAND_DISPLAY XDG_SESSION_TYPE XDG_CURRENT_DESKTOP NIRI_SOCKET
+    systemctl --user unset-environment WAYLAND_DISPLAY DISPLAY XDG_SESSION_TYPE XDG_CURRENT_DESKTOP NIRI_SOCKET
 elif hash dinitctl >/dev/null 2>&1; then
     # Check that the user dinit daemon is running
     if ! pgrep -u "$(id -u)" dinit >/dev/null 2>&1; then