diff --git a/modules/vscodium.nix b/modules/vscodium.nix
index bcc79c6..320bab0 100644
--- a/modules/vscodium.nix
+++ b/modules/vscodium.nix
@@ -1,7 +1,8 @@
-{ config, pkgs, ... }:
+{ config, pkgs, lib, ... }:
 {
   # Integrated Development Environment (IDE)
-  # Wayland support: NIXOS_OZONE_WL=1
+
+  environment.sessionVariables.NIXOS_OZONE_WL = lib.mkIf (config.services.xserver.displayManager.gdm.wayland) "1";
 
   environment.systemPackages = with pkgs; [
     (vscode-with-extensions.override {