diff --git a/modules/system/desktop/default.nix b/modules/system/desktop/default.nix
index 29f56ca..d6fc08b 100644
--- a/modules/system/desktop/default.nix
+++ b/modules/system/desktop/default.nix
@@ -1,13 +1,9 @@
 {
   self,
-  inputs,
+  moduleWithSystem,
   ...
 }: {
-  flake.nixosModules.desktop = {
-    pkgs,
-    lib,
-    ...
-  }: let
+  flake.nixosModules.desktop = moduleWithSystem ({...}: let
     modules = with self.nixosModules; [
       core
       hyprland
@@ -17,5 +13,5 @@
     ];
   in {
     imports = modules;
-  };
+  });
 }