diff --git a/lib/builders.nix b/lib/builders.nix
index a2c11a9..d35487e 100644
--- a/lib/builders.nix
+++ b/lib/builders.nix
@@ -1,7 +1,7 @@
 { nixpkgs, ... }@inputs:
 
 let
-  lib = import ../lib nixpkgs;
+  lib = nixpkgs.lib.extend (import ../lib);
 in
 rec {
   toPartialNixosConfig =
diff --git a/lib/default.nix b/lib/default.nix
index 0cfa6c6..a5c4c3c 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -1,5 +1,5 @@
-lib:
-with lib;
+final: prev:
+with prev;
 rec {
   # Ternary operator
   # Exaample: