diff --git a/flake.nix b/flake.nix
index c48d52b..50522b5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -18,7 +18,6 @@
   outputs = inputs @ {
     self,
     nixpkgs,
-    niri-flake,
     home-manager,
     ...
   }: {
@@ -33,9 +32,6 @@
             useUserPackages = true;
             useGlobalPkgs = true;
             users.benag = ./home/home.nix;
-            sharedModules = [
-              niri-flake.homeModules.niri
-            ];
           };
         }
       ];
diff --git a/home/modules/niri.nix b/home/modules/niri.nix
index ea3dba9..d39b7c2 100644
--- a/home/modules/niri.nix
+++ b/home/modules/niri.nix
@@ -1,4 +1,6 @@
-{pkgs, ...}: {
+{inputs, ...}: {
+  imports = [inputs.niri-flake.homeModules.niri];
+
   programs.niri = {
     enable = true;