diff --git a/python/helpers/pytest_teamcity.py b/python/helpers/pytest_teamcity.py
index 76081d6d99a1..c1b69f5ad619 100644
--- a/python/helpers/pytest_teamcity.py
+++ b/python/helpers/pytest_teamcity.py
@@ -63,6 +63,8 @@ if PYVERSION > [1,3,4]:
     messages.testFinished(name)
 
   def pytest_sessionfinish(session, exitstatus):
+    if current_suite:
+      messages.testSuiteFinished(current_suite)
     if current_file_suite:
       messages.testSuiteFinished(current_file_suite)