diff --git a/src/extension.ts b/src/extension.ts
index 7e75948..d705af4 100644
--- a/src/extension.ts
+++ b/src/extension.ts
@@ -163,6 +163,7 @@ export async function activate(context: vscode.ExtensionContext) {
         // If disabled, clear any existing diagnostics for this doc.
         if (!isEnabled) {
             diagnosticCollection.delete(document.uri);
+            documentHashMemory[document.fileName] = '';
             return;
         }
 
@@ -203,6 +204,7 @@ export async function activate(context: vscode.ExtensionContext) {
     // Clean up diagnostics when a file is closed
     vscode.workspace.onDidCloseTextDocument((document: vscode.TextDocument) => {
         diagnosticCollection.delete(document.uri);
+        documentHashMemory[document.fileName] = '';
     }, null, context.subscriptions);
 }