diff --git a/nixos/lib/test-driver/nixos-test-driver-docstrings.nix b/nixos/lib/test-driver/nixos-test-driver-docstrings.nix
index b6450685e047..107d590e2745 100644
--- a/nixos/lib/test-driver/nixos-test-driver-docstrings.nix
+++ b/nixos/lib/test-driver/nixos-test-driver-docstrings.nix
@@ -11,6 +11,6 @@ in
 
 runCommand "nixos-test-driver-docstrings" env ''
   mkdir $out
-  python3 ${./src/extract-docstrings.py} ${./src/test_driver/machine.py} \
+  python3 ${./src/extract-docstrings.py} ${./src/test_driver/machine/__init__.py} \
     > $out/machine-methods.md
 ''
diff --git a/nixos/lib/test-driver/src/test_driver/machine.py b/nixos/lib/test-driver/src/test_driver/machine/__init__.py
similarity index 100%
rename from nixos/lib/test-driver/src/test_driver/machine.py
rename to nixos/lib/test-driver/src/test_driver/machine/__init__.py
diff --git a/nixos/lib/test-driver/src/test_driver/qmp.py b/nixos/lib/test-driver/src/test_driver/machine/qmp.py
similarity index 100%
rename from nixos/lib/test-driver/src/test_driver/qmp.py
rename to nixos/lib/test-driver/src/test_driver/machine/qmp.py