diff --git a/src/cleveragents/infrastructure/database/unit_of_work.py b/src/cleveragents/infrastructure/database/unit_of_work.py
index 0b9f4c491..693048285 100644
--- a/src/cleveragents/infrastructure/database/unit_of_work.py
+++ b/src/cleveragents/infrastructure/database/unit_of_work.py
@@ -163,6 +163,7 @@ class UnitOfWork:
         if self._engine is not None and self.database_url != "sqlite:///:memory:":
             self._engine.dispose()
             self._engine = None
+            self._session_factory = None
 
 
 class UnitOfWorkContext: