diff --git a/vyper_lsp/main.py b/vyper_lsp/main.py
index 88c1af2..5915f7d 100755
--- a/vyper_lsp/main.py
+++ b/vyper_lsp/main.py
@@ -17,6 +17,7 @@
     CompletionParams,
     CompletionList,
     DeclarationParams,
+    ReferenceParams,
     DefinitionParams,
     HoverParams,
     Hover,
@@ -130,7 +131,7 @@ def go_to_definition(
 
 
 @server.feature(TEXT_DOCUMENT_REFERENCES)
-def find_references(ls: LanguageServer, params: DefinitionParams) -> List[Location]:
+def find_references(ls: LanguageServer, params: ReferenceParams) -> List[Location]:
     document = ls.workspace.get_text_document(params.text_document.uri)
     return [
         Location(uri=params.text_document.uri, range=range_)