diff --git a/libs/core/langchain_core/retrievers.py b/libs/core/langchain_core/retrievers.py
index d1a5cb8cf93..47b6b2b6167 100644
--- a/libs/core/langchain_core/retrievers.py
+++ b/libs/core/langchain_core/retrievers.py
@@ -88,7 +88,8 @@ class BaseRetriever(RunnableSerializable[RetrieverInput, RetrieverOutput], ABC):
 
         .. code-block:: python
 
-            from langchain_core import Document, BaseRetriever
+            from langchain_core.documents import Document
+            from langchain_core.retrievers import BaseRetriever
             from typing import List
 
             class SimpleRetriever(BaseRetriever):