diff --git a/lib/default.nix b/lib/default.nix
index 7fd6a5c6..cbb37010 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -6,10 +6,12 @@
 # If you have to add those functions, you'll have to add them in configUtils.
 { pkgs }:
 
+let
+  inherit (pkgs) lib;
+in
 pkgs.lib.makeExtensible
 (self:
   let
-    inherit (pkgs) lib;
     callLib = file: import file { inherit pkgs lib self; };
   in {
     builders = callLib ./builders.nix;
@@ -19,4 +21,8 @@ pkgs.lib.makeExtensible
     inherit (self.builders) makeXDGMimeAssociationList makeXDGPortalConfiguration;
     inherit (self.trivial) countAttrs;
     inherit (self.data) importYAML renderTeraTemplate;
+  } // lib.optionalAttrs (builtins ? fetchTree) {
+    flake = callLib ./flake.nix;
+
+    inherit (self.flake) importFlakeMetadata fetchTree fetchInput;
   })