diff --git a/examples/python/simple_example.py b/examples/python/simple_example.py
index c7d12e5ac..47940ca6e 100644
--- a/examples/python/simple_example.py
+++ b/examples/python/simple_example.py
@@ -1,5 +1,6 @@
-import cognee
 import asyncio
+
+import cognee
 from cognee.api.v1.search import SearchType
 
 # Prerequisites:
@@ -7,6 +8,7 @@ from cognee.api.v1.search import SearchType
 # 2. Add your OpenAI API key to the `.env` file in the `LLM_API_KEY` field:
 #    LLM_API_KEY = "your_key_here"
 
+
 async def main():
     # Create a clean slate for cognee -- reset data and system state
     await cognee.prune.prune_data()
@@ -26,7 +28,7 @@ async def main():
 
     # Query cognee for insights on the added text
     search_results = await cognee.search(
-        SearchType.INSIGHTS, query='Tell me about NLP'
+        SearchType.INSIGHTS, query_text='Tell me about NLP'
     )
 
     # Display search results