diff --git a/modules/base/system/default.nix b/modules/base/system/default.nix
index 733c1c2..0f6c9fc 100644
--- a/modules/base/system/default.nix
+++ b/modules/base/system/default.nix
@@ -30,6 +30,7 @@
         {
           imports = [
             inputs.sops-nix.nixosModules.sops
+            inputs.niri-flake.nixosModules.niri
           ];
 
           environment.systemPackages = with pkgs; [
diff --git a/modules/desktop/niri.nix b/modules/desktop/niri.nix
index 505e118..1455b82 100644
--- a/modules/desktop/niri.nix
+++ b/modules/desktop/niri.nix
@@ -3,9 +3,6 @@
     nixos.niri =
       { inputs, pkgs, ... }:
       {
-        imports = [
-          inputs.niri-flake.nixosModules.niri
-        ];
 
         programs.niri = {
           enable = true;