Apply the wrapper to the default vm

lib/config/custom-build-vm.nix

@@ -1,25 +1,45 @@
-{ config, lib, pkgs, ... }:
+{ config, lib, options, pkgs, ... }: {
-let
+  options = {};
-  createVmRunner =
+
-    vmVariant:
-      pkgs.writeShellApplication {
-        name = "run-${config.system.name}-vm";
-        text = ''
-          ${vmVariant.system.build.vm}/bin/run-${config.system.name}-vm
-        '';
-      };
-in
-  {
   config = {
     # Replace native `qemu` with `remote-viewer`
     system.build =
-        with {
-          inherit (config.virtualisation)
-            vmVariantWithBootLoader
-          ;
-        };
       {
-          vmWithBootLoader = createVmRunner vmVariantWithBootLoader;
+        vm =
+          let
+            packageName = "custom-nixos-vm";
+
+            mergedSystem =
+              with options.system;
+              lib.mergeDefinitions
+                build.loc
+                build.type
+                (lib.lists.forEach
+                  (
+                    builtins.filter
+                      (item:
+                        !(lib.path.hasPrefix ./. (/. + item.file)))
+                      build.definitionsWithLocations)
+                  (item: { inherit (item) file value; }));
+          in
+            lib.mkForce (
+              if (mergedSystem.mergedValue.vm.name == packageName)
+              then
+                mergedSystem.mergedValue.vm
+              else
+                let
+                  wrapped = pkgs.writeShellApplication {
+                    name = "run-${config.system.name}-vm";
+                    text = ''
+                      ${mergedSystem.mergedValue.vm}/bin/run-${config.system.name}-vm
+                    '';
+                  };
+                in
+                  # Rename package to `nixos-vm`
+                  pkgs.symlinkJoin {
+                    name = packageName;
+                    paths = [ wrapped ];
+                  });
       };
 
     virtualisation =
@@ -43,4 +63,4 @@ in
         vmVariantWithBootLoader = vmConfig;
       };
   };
-  }
+}