diff --git a/tests/test_docs.py b/tests/test_docs.py
index 61806de9..ea3d92d9 100644
--- a/tests/test_docs.py
+++ b/tests/test_docs.py
@@ -88,6 +88,10 @@ def poll_workspace(config, stop_event):
                     env = {
                         **os.environ,
                         "SCHEMA_PATH": str(config["schema_path"].resolve()),
+                        "PYTHONPATH": ":".join(
+                            os.environ.get("PYTHONPATH", "").split(":")
+                            + [str(pathlib.Path(__file__).parents[1].resolve())]
+                        ),
                     }
                     exit_code = 0
                     try: