diff --git a/flake.nix b/flake.nix
index 1a68d43..746c210 100644
--- a/flake.nix
+++ b/flake.nix
@@ -59,12 +59,13 @@
             (nixpkgs.lib.optionals (class == "nixos") [
               home-manager.nixosModules.home-manager
               agenix.nixosModules.default
-              stylix.nixosModules.default
+              stylix.nixosModules.stylix
               niri-flake.nixosModules.niri
             ])
             ++ (nixpkgs.lib.optionals (class == "darwin") [
               home-manager.darwinModules.home-manager
               agenix.darwinModules.default
+              stylix.darwinModules.stylix
             ]);
         };