diff --git a/install-cli.nix b/install-cli.nix
index 1f52419..7deb16c 100644
--- a/install-cli.nix
+++ b/install-cli.nix
@@ -8,7 +8,7 @@ let
   originalSystem = (builtins.getFlake "${flake}").nixosConfigurations."${flakeAttr}";
   diskoSystem =
     let
-      lib = originalSystem.lib;
+      lib = originalSystem.pkgs.lib;
 
       modifiedDisks = builtins.mapAttrs
         (name: value: let