diff --git a/mem0/memory/main.py b/mem0/memory/main.py
index 8e2c47e9..951539a8 100644
--- a/mem0/memory/main.py
+++ b/mem0/memory/main.py
@@ -377,6 +377,14 @@ class Memory(MemoryBase):
         new_metadata["data"] = data
         new_metadata["created_at"] = existing_memory.payload.get("created_at")
         new_metadata["updated_at"] = datetime.now(pytz.timezone('US/Pacific')).isoformat()
+        
+        if "user_id" in existing_memory.payload: 
+            new_metadata["user_id"] = existing_memory.payload["user_id"]
+        if "agent_id" in existing_memory.payload:
+            new_metadata["agent_id"] = existing_memory.payload["agent_id"]
+        if "run_id" in existing_memory.payload:
+            new_metadata["run_id"] = existing_memory.payload["run_id"]
+               
         embeddings = self.embedding_model.embed(data)
         self.vector_store.update(
             name=self.collection_name,