diff --git a/pythonFiles/vscode_pytest/run_pytest_script.py b/pythonFiles/vscode_pytest/run_pytest_script.py
index 57bb41a3a7dd..ffb4d0c55b16 100644
--- a/pythonFiles/vscode_pytest/run_pytest_script.py
+++ b/pythonFiles/vscode_pytest/run_pytest_script.py
@@ -21,6 +21,7 @@
     # Add the root directory to the path so that we can import the plugin.
     directory_path = pathlib.Path(__file__).parent.parent
     sys.path.append(os.fspath(directory_path))
+    sys.path.insert(0, os.getcwd())
     # Get the rest of the args to run with pytest.
     args = sys.argv[1:]
     run_test_ids_port = os.environ.get("RUN_TEST_IDS_PORT")