1 files changed, 8 insertions, 11 deletions
diff --git a/extension/src/lang-server/codeLens.ts b/extension/src/lang-server/codeLens.ts
index 21448e31..3bd4f153 100644
--- a/extension/src/lang-server/codeLens.ts
+++ b/extension/src/lang-server/codeLens.ts
@@ -60,18 +60,15 @@ class SuggestionsCodeLensProvider implements vscode.CodeLensProvider {
   }
 }
 
-const allCodeLensProviders: { [langauge: string]: vscode.CodeLensProvider[] } =
-  {
-    // python: [new SuggestionsCodeLensProvider(), new PytestCodeLensProvider()],
-    "*": [new SuggestionsCodeLensProvider()],
-  };
+let suggestionsCodeLensDisposable: vscode.Disposable | undefined = undefined;
 
 export function registerAllCodeLensProviders(context: vscode.ExtensionContext) {
-  for (const language in allCodeLensProviders) {
-    for (const codeLensProvider of allCodeLensProviders[language]) {
-      context.subscriptions.push(
-        vscode.languages.registerCodeLensProvider(language, codeLensProvider)
-      );
-    }
+  if (suggestionsCodeLensDisposable) {
+    suggestionsCodeLensDisposable.dispose();
   }
+  suggestionsCodeLensDisposable = vscode.languages.registerCodeLensProvider(
+    "*",
+    new SuggestionsCodeLensProvider()
+  );
+  context.subscriptions.push(suggestionsCodeLensDisposable);
 }