diff --git a/modules/system/boot/stage-2.nix b/modules/system/boot/stage-2.nix
index 69c85d43bc5..f0e90452ebe 100644
--- a/modules/system/boot/stage-2.nix
+++ b/modules/system/boot/stage-2.nix
@@ -46,9 +46,5 @@ in
 {
   require = [options];
 
-  system = {
-    build = {
-      inherit bootStage2;
-    };
-  };
+  system.build.bootStage2 = bootStage2;
 }