diff --git a/nix/pkgs/default.nix b/nix/pkgs/default.nix
index 1573b90..6449b7b 100644
--- a/nix/pkgs/default.nix
+++ b/nix/pkgs/default.nix
@@ -25,7 +25,7 @@ let
     ) (builtins.attrNames config.site.hosts)
   );
 
-  network-graphs = pkgs.callPackage ./network-graphs.nix { inherit config; };
+  network-graphs = import ./network-graphs.nix { inherit config pkgs; };
 
   mkRootfs = hostName:
     self.nixosConfigurations.${hostName}.config.system.build.toplevel;
diff --git a/nix/pkgs/network-graphs.nix b/nix/pkgs/network-graphs.nix
index c3c33c0..ea5dde9 100644
--- a/nix/pkgs/network-graphs.nix
+++ b/nix/pkgs/network-graphs.nix
@@ -1,5 +1,7 @@
-{ lib, config, runCommand, graphviz, ... }:
+{ config, pkgs, ... }:
 let
+  inherit (pkgs) lib runCommand graphviz;
+
   netColor = net:
     if net == "core"
     then "grey"