diff --git a/docs/pydoc/config/agent.yml b/docs/pydoc/config/agent.yml
index 75befef7e..b6b56bf47 100644
--- a/docs/pydoc/config/agent.yml
+++ b/docs/pydoc/config/agent.yml
@@ -1,7 +1,7 @@
 loaders:
   - type: python
-    search_path: [../../../haystack/agents/agent]
-    modules: ['agent']
+    search_path: [../../../haystack/agents]
+    modules: ['base']
     ignore_when_discovered: ['__init__']
 processors:
   - type: filter