From bdff14c1df19957b9f38f8d57752a2ac2b08f811 Mon Sep 17 00:00:00 2001
From: Heyward Fann <fannheyward@gmail.com>
Date: Tue, 18 Nov 2025 14:36:51 +0800
Subject: [PATCH] fix(language-client): correct unregister

---
 src/language-client/semanticTokens.ts      | 1 +
 src/language-client/textSynchronization.ts | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/language-client/semanticTokens.ts b/src/language-client/semanticTokens.ts
index d3294a94729..3122ea7ed68 100644
--- a/src/language-client/semanticTokens.ts
+++ b/src/language-client/semanticTokens.ts
@@ -70,6 +70,7 @@ export class SemanticTokensFeature extends TextDocumentLanguageFeature<boolean |
       SemanticTokenTypes.number,
       SemanticTokenTypes.regexp,
       SemanticTokenTypes.decorator,
+      SemanticTokenTypes.label,
       SemanticTokenTypes.operator
     ]
     capability.tokenModifiers = [
diff --git a/src/language-client/textSynchronization.ts b/src/language-client/textSynchronization.ts
index d60ad10334c..a918e645b5e 100644
--- a/src/language-client/textSynchronization.ts
+++ b/src/language-client/textSynchronization.ts
@@ -217,7 +217,7 @@ export class DidCloseTextDocumentFeature extends TextDocumentEventFeature<DidClo
   }
 
   public unregister(id: string): void {
-    let selector = this._selectors.get(id)!
+    let selector = this._selectors.get(id)
     if (!selector) return
     // The super call removed the selector from the map
     // of selectors.