Refactor server lifecycle

1 files changed, 3 insertions, 1 deletions

diff --git a/editors/code/src/status_display.ts b/editors/code/src/status_display.ts
index e3719075b..1454bf8b0 100644
--- a/editors/code/src/status_display.ts
+++ b/editors/code/src/status_display.ts
@@ -7,7 +7,9 @@ const spinnerFrames = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '
 export function activateStatusDisplay(ctx: Ctx) {
     const statusDisplay = new StatusDisplay(ctx.config.cargoWatchOptions.command);
     ctx.pushCleanup(statusDisplay);
-    ctx.onNotification('$/progress', params => statusDisplay.handleProgressNotification(params));
+    ctx.onDidRestart(client => {
+        client.onNotification('$/progress', params => statusDisplay.handleProgressNotification(params));
+    })
 }
 
 class StatusDisplay implements vscode.Disposable {