diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py
index 584f92832..b5122b164 100644
--- a/spacy/ml/models/entity_linker.py
+++ b/spacy/ml/models/entity_linker.py
@@ -130,7 +130,7 @@ def get_candidates(kb: KnowledgeBase, mention: Span) -> Iterable[Candidate]:
 
 
 def get_candidates_batch(
-    kb: KnowledgeBase, mentions: Iterable[SpanGroup]
+    kb: KnowledgeBase, mentions: SpanGroup
 ) -> Iterable[Iterable[Candidate]]:
     """
     Return candidate entities for the given mentions and fetching appropriate entries from the index.
diff --git a/website/docs/api/inmemorylookupkb.mdx b/website/docs/api/inmemorylookupkb.mdx
index 87e8e97fe..8852f4b16 100644
--- a/website/docs/api/inmemorylookupkb.mdx
+++ b/website/docs/api/inmemorylookupkb.mdx
@@ -197,7 +197,7 @@ to you.
 
 | Name        | Description                                                                                                  |
 | ----------- | ------------------------------------------------------------------------------------------------------------ |
-| `mentions`  | The textual mentions. ~~Iterable[SpanGroup]~~                                                             |
+| `mentions`  | The textual mentions. ~~SpanGroup~~                                                             |
 | **RETURNS** | An iterable of iterable with relevant `InMemoryCandidate` objects. ~~Iterable[Iterable[InMemoryCandidate]]~~ |
 
 
diff --git a/website/docs/api/kb.mdx b/website/docs/api/kb.mdx
index a0d01c6ad..850c68926 100644
--- a/website/docs/api/kb.mdx
+++ b/website/docs/api/kb.mdx
@@ -101,7 +101,7 @@ to you.
 
 | Name        | Description                                                                                  |
 | ----------- | -------------------------------------------------------------------------------------------- |
-| `mentions`  | The textual mentions. ~~Iterable[SpanGroup]~~                                             |
+| `mentions`  | The textual mentions. ~~SpanGroup~~                                             |
 | **RETURNS** | An iterable of iterable with relevant `Candidate` objects. ~~Iterable[Iterable[Candidate]]~~ |
 
 ## KnowledgeBase.get_vector {id="get_vector",tag="method"}