diff --git a/src/PhpDocument.php b/src/PhpDocument.php
index 051dc96..a23b4de 100644
--- a/src/PhpDocument.php
+++ b/src/PhpDocument.php
@@ -2,7 +2,7 @@
 
 namespace LanguageServer;
 
-use \LanguageServer\Protocol\{Diagnostic, DiagnosticSeverity, Range, Position, SymbolKind};
+use \LanguageServer\Protocol\{Diagnostic, DiagnosticSeverity, Range, Position, SymbolKind, TextEdit};
 
 use PhpParser\{Error, Comment, Node, ParserFactory, NodeTraverser, Lexer, Parser};
 use PhpParser\PrettyPrinter\Standard as PrettyPrinter;