diff --git a/docs/examples/customer-support-agent.mdx b/docs/examples/customer-support-agent.mdx
index 74ec95638..9fb3f4162 100644
--- a/docs/examples/customer-support-agent.mdx
+++ b/docs/examples/customer-support-agent.mdx
@@ -23,6 +23,7 @@ pip install openai mem0ai
 Below is the simplified code to create and interact with a Customer Support AI Agent using Mem0:
 
 ```python
+import os
 from openai import OpenAI
 from mem0 import Memory
 
diff --git a/docs/examples/mem0-with-ollama.mdx b/docs/examples/mem0-with-ollama.mdx
index a4257ae26..556eeb937 100644
--- a/docs/examples/mem0-with-ollama.mdx
+++ b/docs/examples/mem0-with-ollama.mdx
@@ -21,7 +21,6 @@ Before you begin, ensure you have Mem0 and Ollama installed and properly configu
 Below is the complete code to set up and use Mem0 locally with Ollama:
 
 ```python
-import os
 from mem0 import Memory
 
 config = {