diff --git a/docs/examples/custom_storage_knowledge_source_example.py b/docs/examples/custom_storage_knowledge_source_example.py
index b592e4810..58ec84fe0 100644
--- a/docs/examples/custom_storage_knowledge_source_example.py
+++ b/docs/examples/custom_storage_knowledge_source_example.py
@@ -2,6 +2,7 @@
 
 import chromadb
 from chromadb.config import Settings
+
 from crewai import Agent, Crew, Task
 from crewai.knowledge.source.custom_storage_knowledge_source import CustomStorageKnowledgeSource
 from crewai.knowledge.storage.knowledge_storage import KnowledgeStorage
@@ -56,7 +57,7 @@ def main():
     knowledge_source = get_knowledge_source_with_custom_storage(folder_name="example")
     
     agent = Agent(role="test", goal="test", backstory="test")
-    task = Task(description="test", agent=agent)
+    task = Task(description="test", expected_output="test", agent=agent)
     
     crew = Crew(
         agents=[agent],