diff --git a/nixos/lib/testing-python.nix b/nixos/lib/testing-python.nix
index 6369d6ef05d4..070a3a7ddc21 100644
--- a/nixos/lib/testing-python.nix
+++ b/nixos/lib/testing-python.nix
@@ -186,6 +186,12 @@ rec {
           --set startScripts "''${vmStartScripts[*]}" \
           --set testScript "$out/test-script" \
           --set vlans '${toString vlans}'
+
+        ln -s ${testDriver}/bin/nixos-test-driver $out/bin/nixos-run-vms
+        wrapProgram $out/bin/nixos-run-vms \
+          --set startScripts "''${vmStartScripts[*]}" \
+          --set testScript "${pkgs.writeText "start-all" "start_all(); join_all();"}" \
+          --set vlans '${toString vlans}'
       '');
 
   # Make a full-blown test