diff --git a/src/session.sh b/src/session.sh
index 0e991f1..24b9606 100755
--- a/src/session.sh
+++ b/src/session.sh
@@ -38,8 +38,10 @@
 #  - https://systemd.io/DESKTOP_ENVIRONMENTS/
 #
 export XDG_CURRENT_DESKTOP=sway
+export XDG_SESSION_DESKTOP=${XDG_SESSION_DESKTOP:-sway}
 export XDG_SESSION_TYPE=wayland
-VARIABLES="DISPLAY I3SOCK SWAYSOCK WAYLAND_DISPLAY XDG_CURRENT_DESKTOP XDG_SESSION_TYPE"
+VARIABLES="DESKTOP_SESSION XDG_CURRENT_DESKTOP XDG_SESSION_DESKTOP XDG_SESSION_TYPE"
+VARIABLES="${VARIABLES} DISPLAY I3SOCK SWAYSOCK WAYLAND_DISPLAY"
 SESSION_TARGET="sway-session.target"
 WITH_CLEANUP=""