diff --git a/nixos/lib/test-driver/src/test_driver/__init__.py b/nixos/lib/test-driver/src/test_driver/__init__.py
index 20f668867da6..1e5a062b18c8 100644
--- a/nixos/lib/test-driver/src/test_driver/__init__.py
+++ b/nixos/lib/test-driver/src/test_driver/__init__.py
@@ -174,7 +174,11 @@ def generate_driver_symbols() -> None:
             vlans=[],
             global_timeout=0,
             enable_ssh_backdoor=False,
-            test_script=Path("testScriptWithTypes"),
+            test_script=(
+                Path("testScriptWithTypes")
+                if (Path("testScriptWithTypes").is_file())
+                else Path("testScriptFile")
+            ),
         ),
         out_dir=Path(),
         logger=CompositeLogger([]),
diff --git a/nixos/lib/testing/driver.nix b/nixos/lib/testing/driver.nix
index f49f1fe199b1..c6802d61dd43 100644
--- a/nixos/lib/testing/driver.nix
+++ b/nixos/lib/testing/driver.nix
@@ -85,6 +85,8 @@ let
                 testScriptWithTypes
         ''}
 
+        echo -n "$testScript" > testScriptFile
+
         cp "${config.driverConfiguration.test_script}" $out/test-script
 
         ln -s ${testDriver}/bin/nixos-test-driver $out/bin/nixos-test-driver