diff --git a/pythonFiles/vscode_pytest/__init__.py b/pythonFiles/vscode_pytest/__init__.py
index 1718d435bb23..0767b85c5249 100644
--- a/pythonFiles/vscode_pytest/__init__.py
+++ b/pythonFiles/vscode_pytest/__init__.py
@@ -235,6 +235,7 @@ def pytest_report_teststatus(report, config):
 }
 
 
+@pytest.hookimpl(hookwrapper=True, trylast=True)
 def pytest_runtest_protocol(item, nextitem):
     map_id_to_path[item.nodeid] = get_node_path(item)
     skipped = check_skipped_wrapper(item)
@@ -257,6 +258,7 @@ def pytest_runtest_protocol(item, nextitem):
                 "success",
                 collected_test if collected_test else None,
             )
+    yield
 
 
 def check_skipped_wrapper(item):