diff --git a/packages/lib-doc/default.nix b/packages/lib-docs/default.nix
similarity index 77%
rename from packages/lib-doc/default.nix
rename to packages/lib-docs/default.nix
index 1c27e9b..b5f04ca 100644
--- a/packages/lib-doc/default.nix
+++ b/packages/lib-docs/default.nix
@@ -1,8 +1,13 @@
 # Generates the documentation for library functions via nixdoc.
-{ pkgs, inputs, ... }:
+{
+  pkgs,
+  inputs,
+  lib,
+  ...
+}:
 let
 
-  inherit (pkgs) stdenv lib nixdoc;
+  inherit (pkgs) stdenv nixdoc;
 
 in
 stdenv.mkDerivation {
@@ -17,9 +22,9 @@ stdenv.mkDerivation {
       description=$3
       # TODO: wrap lib.$name in <literal>, make nixdoc not escape it
       if [[ -e "../lib/$baseName.nix" ]]; then
-        nixdoc -c "$name" -d "lib.$name: $description" -l ${pkgs.doc-locations-json} -f "$baseName.nix" > "$out/$name.md"
+        nixdoc -c "$name" -d "lib.$name: $description" -l ${pkgs.amg.doc-locations-json} -f "$baseName.nix" > "$out/$name.md"
       else
-        nixdoc -c "$name" -d "lib.$name: $description" -l ${pkgs.doc-locations-json} -f "$baseName/default.nix" > "$out/$name.md"
+        nixdoc -c "$name" -d "lib.$name: $description" -l ${pkgs.amg.doc-locations-json} -f "$baseName/default.nix" > "$out/$name.md"
       fi
     }
 
@@ -34,6 +39,6 @@ stdenv.mkDerivation {
       ''
         docgen ${name} ${baseName} ${lib.escapeShellArg description}
       ''
-    ) pkgs.libsets}
+    ) lib.amg.libsets}
   '';
 }