diff --git a/level_3/models/memory.py b/level_3/models/memory.py
index b326da7e6..e19ecddc4 100644
--- a/level_3/models/memory.py
+++ b/level_3/models/memory.py
@@ -12,12 +12,14 @@ class MemoryModel(Base):
 
     id = Column(String, primary_key=True)
     user_id = Column(String, ForeignKey('users.id'), index=True)
+    operation_id = Column(String, ForeignKey('operations.id'), index=True)
     created_at = Column(DateTime, default=datetime.utcnow)
     updated_at = Column(DateTime, onupdate=datetime.utcnow)
     methods_list = Column(String , nullable=True)
     attributes_list = Column(String, nullable=True)
 
     user = relationship("User", back_populates="memories")
+    operation = relationship("Operation", back_populates="memories")
     metadatas = relationship("MetaDatas", back_populates="memory", cascade="all, delete-orphan")
 
     def __repr__(self):
diff --git a/level_3/rag_test_manager.py b/level_3/rag_test_manager.py
index 47f07269a..edb0ed787 100644
--- a/level_3/rag_test_manager.py
+++ b/level_3/rag_test_manager.py
@@ -300,7 +300,7 @@ async def start_test(data, test_set=None, user_id=None, params=None, job_id=None
 
         if test_set_id is None:
             test_set_id = str(uuid.uuid4())
-            await add_entity(session, TestSet(id=test_set_id, user_id=user_id, content=str(test_set)))
+            await add_entity(session, TestSet(id=test_set_id, user_id=user_id,  content=str(test_set)))
 
         if params is None:
             data_format = data_format_route(data)  # Assume data_format_route is predefined
@@ -412,7 +412,7 @@ async def start_test(data, test_set=None, user_id=None, params=None, job_id=None
         for result_list in results[0]:
             for result in result_list:
                 print("Here is one result", result)
-                await add_entity(session, TestOutput(id=test_id ,set_id=str(uuid.uuid4()), user_id=user_id, test_results=result['success'], test_score=str(result['score']), test_metric_name=result['metric_name'], test_query=result['query'], test_output=result['output'], test_expected_output=str(['expected_output']), test_context=result['context'][0]))
+                await add_entity(session, TestOutput(id=test_id , test_set_id=test_set_id, operation_id=job_id,  set_id=str(uuid.uuid4()), user_id=user_id, test_results=result['success'], test_score=str(result['score']), test_metric_name=result['metric_name'], test_query=result['query'], test_output=result['output'], test_expected_output=str(['expected_output']), test_context=result['context'][0]))
 
         print(results)