diff --git a/lib/default.nix b/lib/default.nix
index 1377c8b..9a01a2f 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -407,7 +407,7 @@ let
               ${cfg.config._create}
             '';
 
-            system.build.mountScriptNoDeps = (diskoLib.writeCheckedBash { inherit pkgs checked; noDeps = true; }) "disko-mount" ''
+            mountScriptNoDeps = (diskoLib.writeCheckedBash { inherit pkgs checked; noDeps = true; }) "disko-mount" ''
               ${cfg.config._mount}
             '';