diff --git a/src/PowerShellEditorServices/Services/PowerShell/Host/PsesInternalHost.cs b/src/PowerShellEditorServices/Services/PowerShell/Host/PsesInternalHost.cs
index da263e698..16a7dcc49 100644
--- a/src/PowerShellEditorServices/Services/PowerShell/Host/PsesInternalHost.cs
+++ b/src/PowerShellEditorServices/Services/PowerShell/Host/PsesInternalHost.cs
@@ -168,12 +168,12 @@ public override void ExitNestedPrompt()
 
         public override void NotifyBeginApplication()
         {
-            // TODO: Work out what to do here
+            _hostInfo.PSHost.NotifyBeginApplication();
         }
 
         public override void NotifyEndApplication()
         {
-            // TODO: Work out what to do here
+            _hostInfo.PSHost.NotifyEndApplication();
         }
 
         public void PopRunspace()