diff --git a/nixos/lib/testing/network.nix b/nixos/lib/testing/network.nix
index 513ec7a14092..04ea9a2bc9f7 100644
--- a/nixos/lib/testing/network.nix
+++ b/nixos/lib/testing/network.nix
@@ -112,6 +112,6 @@ let
 in
 {
   config = {
-    defaults = { imports = [ networkModule nodeNumberModule ]; };
+    extraBaseModules = { imports = [ networkModule nodeNumberModule ]; };
   };
 }
diff --git a/nixos/lib/testing/nodes.nix b/nixos/lib/testing/nodes.nix
index c2d4ad6e61bc..765af2878dfe 100644
--- a/nixos/lib/testing/nodes.nix
+++ b/nixos/lib/testing/nodes.nix
@@ -22,6 +22,7 @@ let
               # when the test framework is used by Nixpkgs NixOS tests.
               nixpkgs.config.allowAliases = false;
             })
+          testModuleArgs.config.extraBaseModules
         ] ++ optional config.minimal ../../modules/testing/minimal-kernel.nix;
     };