diff --git a/nixos/lib/test-driver/test-driver.py b/nixos/lib/test-driver/test-driver.py
index 488789e119d0..f8502188bde8 100755
--- a/nixos/lib/test-driver/test-driver.py
+++ b/nixos/lib/test-driver/test-driver.py
@@ -994,8 +994,10 @@ def _test_symbols() -> Dict[str, Any]:
         subtest=subtest,
         run_tests=run_tests,
         join_all=join_all,
+        retry=retry,
         serial_stdout_off=serial_stdout_off,
         serial_stdout_on=serial_stdout_on,
+        Machine=Machine,  # for typing
     )
     return general_symbols
 
diff --git a/nixos/tests/nsd.nix b/nixos/tests/nsd.nix
index 7387f4f1dfa1..eea5a82f6f92 100644
--- a/nixos/tests/nsd.nix
+++ b/nixos/tests/nsd.nix
@@ -85,6 +85,7 @@ in import ./make-test-python.nix ({ pkgs, ...} : {
         self = clientv4 if type == 4 else clientv6
         out = self.succeed(f"host -{type} -t {rr} {query}").rstrip()
         self.log(f"output: {out}")
+        import re
         assert re.search(
             expected, out
         ), f"DNS IPv{type} query on {query} gave '{out}' instead of '{expected}'"