diff --git a/website/docs/api/kb_in_memory.md b/website/docs/api/kb_in_memory.md
index c9ce624f0..9e3279e6a 100644
--- a/website/docs/api/kb_in_memory.md
+++ b/website/docs/api/kb_in_memory.md
@@ -21,9 +21,9 @@ Create the knowledge base.
 > #### Example
 >
 > ```python
-> from spacy.kb import KnowledgeBase
+> from spacy.kb import InMemoryLookupKB
 > vocab = nlp.vocab
-> kb = KnowledgeBase(vocab=vocab, entity_vector_length=64)
+> kb = InMemoryLookupKB(vocab=vocab, entity_vector_length=64)
 > ```
 
 | Name                   | Description                                      |