diff --git a/orchestration/magic_plan_orchestrator.py b/orchestration/magic_plan_orchestrator.py
index 1ad84b81..8b48380e 100644
--- a/orchestration/magic_plan_orchestrator.py
+++ b/orchestration/magic_plan_orchestrator.py
@@ -59,7 +59,9 @@ class MagicPlanService:
         with db_session() as session:
             session.add(uploaded_file)
             session.flush()
-            MagicPlanPostgresRepository(session).save(plan, cast(int, uploaded_file.id))
+            MagicPlanPostgresRepository(session).save(
+                plan, cast(int, uploaded_file.id)
+            )  # TODO: refactor to use postgres Unit of Work
 
         return plan