diff --git a/flake.nix b/flake.nix
index 9b5d11c..35fb627 100644
--- a/flake.nix
+++ b/flake.nix
@@ -84,6 +84,11 @@
           inputs.stylix.homeManagerModules.stylix
           # inputs.xremap.homeManaggerModules.default
           ./home
+          ({pkgs, ...}: {
+            stylix.enable = true;
+            stylix.base16Scheme = "${pkgs.base16-schemes}/share/themes/catppuccin-mocha.yaml";
+            stylix.image = ../static/guz-wallpaper-default.png;
+          })
         ];
         pkgs = import nixpkgs {system = "x86_64-linux";};
       };
diff --git a/home/desktop.nix b/home/desktop.nix
index 05bd573..7d88a39 100644
--- a/home/desktop.nix
+++ b/home/desktop.nix
@@ -2,20 +2,8 @@
   config,
   pkgs,
   lib,
-  osConfig ? null,
   ...
-}:
-(
-  # Check if it is being used as a NixOS module
-  if (isNull osConfig)
-  then {
-    stylix.enable = true;
-    stylix.base16Scheme = "${pkgs.base16-schemes}/share/themes/catppuccin-mocha.yaml";
-    stylix.image = ../static/guz-wallpaper-default.png;
-  }
-  else {}
-)
-// {
+}: {
   # Desktop Environment
 
   ## Hyprland