diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py
index 5da0544a9..ea8882430 100644
--- a/spacy/ml/models/entity_linker.py
+++ b/spacy/ml/models/entity_linker.py
@@ -136,6 +136,6 @@ def get_candidates_batch(
     Return candidate entities for the given mentions and fetching appropriate entries from the index.
     kb (KnowledgeBase): Knowledge base to query.
     mention (Iterable[Span]): Entity mentions for which to identify candidates.
-    RETURNS (Iterable[Iterable[InMemoryCandidate]]): Identified candidates.
+    RETURNS (Iterable[Iterable[Candidate]]): Identified candidates.
     """
     return kb.get_candidates_batch(mentions)