diff --git a/lib/vscode/src/vs/server/node/connection.ts b/lib/vscode/src/vs/server/node/connection.ts
index 6b7c5ffd..b584a34c 100644
--- a/lib/vscode/src/vs/server/node/connection.ts
+++ b/lib/vscode/src/vs/server/node/connection.ts
@@ -54,7 +54,7 @@ export abstract class Connection {
 export class ManagementConnection extends Connection {
 	public constructor(protected protocol: Protocol, token: string) {
 		super(protocol, token);
-		protocol.onClose(() => this.dispose()); // Explicit close.
+		protocol.onDidDispose(() => this.dispose()); // Explicit close.
 		protocol.onSocketClose(() => this.setOffline()); // Might reconnect.
 	}