diff --git a/memory_module/memory_integration.py b/memory_module/memory_integration.py
index fb4086a..26b8d44 100644
--- a/memory_module/memory_integration.py
+++ b/memory_module/memory_integration.py
@@ -2,6 +2,7 @@ import os
 from typing import List, Dict, Any, Optional
 from datetime import datetime
 import openai
+import threading
 from mem0 import Memory