diff --git a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs
index 11d1980b53..10e4a7379a 100644
--- a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs
+++ b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/Interfaces/IVsTestConsoleWrapper2.cs
@@ -18,6 +18,7 @@ public interface IVsTestConsoleWrapper2 : IVsTestConsoleWrapper
///
/// List of attachements
/// EventHandler to receive session complete event
- void OnTestSessionEnd(IEnumerable attachments, ITestSessionEventsHandler testSessionEventsHandler);
+ void FinalizeMultiTestRuns(IEnumerable attachments, ITestSessionEventsHandler testSessionEventsHandler);
+ // void FinalizeTests(IEnumerable attachments, ITestSessionEventsHandler testSessionEventsHandler);
}
}
diff --git a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs
index 378850eda0..48d44765d5 100644
--- a/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs
+++ b/src/Microsoft.TestPlatform.VsTestConsole.TranslationLayer/VsTestConsoleWrapper.cs
@@ -279,6 +279,7 @@ public void OnTestSessionEnd(IEnumerable attachments, ITestSessio
this.EnsureInitialized();
this.requestSender.OnTestSessionEnd(attachments, testSessionEventsHandler);
+ this.requestSender.FinalizeSession(attachments, testSessionEventsHandler);
}
#endregion