diff --git a/src/tools/ecode/plugins/lsp/lspclientplugin.cpp b/src/tools/ecode/plugins/lsp/lspclientplugin.cpp
index 3cb570fb9..2bcb11361 100644
--- a/src/tools/ecode/plugins/lsp/lspclientplugin.cpp
+++ b/src/tools/ecode/plugins/lsp/lspclientplugin.cpp
@@ -954,7 +954,7 @@ void LSPClientPlugin::onUnregister( UICodeEditor* editor ) {
 	if ( mClosing )
 		return;
 	Lock l( mDocMutex );
-	TextDocument* doc = mEditorDocs[editor];
+	TextDocument* doc = &editor->getDocument();
 	const auto& cbs = mEditors[editor];
 	for ( auto listener : cbs )
 		editor->removeEventListener( listener );