diff --git a/mem0/memory/setup.py b/mem0/memory/setup.py
index 2973b072..a22b2e13 100644
--- a/mem0/memory/setup.py
+++ b/mem0/memory/setup.py
@@ -4,7 +4,7 @@ import uuid
 
 # Set up the directory path
 home_dir = os.path.expanduser("~")
-mem0_dir = os.path.join(home_dir, ".mem0")
+mem0_dir = os.environ.get("MEM0_DIR") or os.path.join(home_dir, ".mem0")
 os.makedirs(mem0_dir, exist_ok=True)