diff --git a/modules/niri.nix b/modules/niri.nix
index 2645df7..85929a3 100644
--- a/modules/niri.nix
+++ b/modules/niri.nix
@@ -13,9 +13,13 @@ in
   config = mkIf cfg.enable {
     programs.niri.enable = true;
 
-    services.displayManager.defaultSession = "niri";
-    services.displayManager.enable = true;
-    services.displayManager.sddm.enable = true;
-    services.displayManager.sddm.wayland.enable = true;
+    environment.sessionVariables.NIXOS_OZONE_WL = "1";
+
+    services.displayManager = {
+      defaultSession = "niri";
+      enable = true;
+      sddm.enable = true;
+      sddm.wayland.enable = true;
+    };
   };
 }