diff --git a/cli.nix b/cli.nix
index 76e0971..f99ad61 100644
--- a/cli.nix
+++ b/cli.nix
@@ -3,8 +3,8 @@
 , diskoFile
 , ... }@args:
 let
-  disko = import ./. {};
-  diskFormat = import diskoFile args;
+  disko = import ./. { inherit (pkgs) lib; };
+  diskFormat = import diskoFile;
   diskoEval = if (mode == "create") then
     disko.createScript diskFormat pkgs
   else if (mode == "mount") then