1 files changed, 7 insertions, 5 deletions

diff --git a/flake.nix b/flake.nix
index 1f51864..9a6dc80 100644
--- a/flake.nix
+++ b/flake.nix
@@ -55,10 +55,12 @@
           }
         )
         systems);
-    mkSystem = hostDir: system:
-      nixpkgs.lib.nixosSystem {
+    mkSystem = hostDir: system: let
+      lib = nixpkgs.lib.extend (_: _: self.lib);
+    in
+      lib.nixosSystem {
         inherit system;
-        specialArgs = {inherit inputs system hostDir;};
+        specialArgs = {inherit inputs system hostDir lib;};
         modules = [
           ./system
           ./options.nix
@@ -67,10 +69,11 @@
       };
     mkHome = hostDir: system: let
       hostModule = mkSystem hostDir system;
+      lib = nixpkgs.lib.extend (_: _: self.lib);
     in
       home-manager.lib.homeManagerConfiguration {
         inherit (hostModule) pkgs;
-        extraSpecialArgs = {inherit inputs system hostDir;};
+        extraSpecialArgs = {inherit inputs system hostDir lib;};
         modules = [
           ./home
           ./options.nix
@@ -97,7 +100,6 @@
     };
 
     lib = import ./lib {
-      inherit inputs;
       inherit (nixpkgs) lib;
     };