diff --git a/src/calculation_target_reliability/router.py b/src/calculation_target_reliability/router.py
index 3158e21..9151ab0 100644
--- a/src/calculation_target_reliability/router.py
+++ b/src/calculation_target_reliability/router.py
@@ -71,7 +71,8 @@ async def get_target_reliability(
         target_eaf=eaf_input,
         db_session=db_session,
         oh_session_id=oh_session_id,
-        collector_db=collector_db
+        collector_db=collector_db,
+        simulation_id=simulation_id
     )
 
 
diff --git a/src/calculation_target_reliability/schema.py b/src/calculation_target_reliability/schema.py
index df02d2c..8fa3890 100644
--- a/src/calculation_target_reliability/schema.py
+++ b/src/calculation_target_reliability/schema.py
@@ -54,6 +54,7 @@ class OptimizationResult(OverhaulBase):
     eaf_gap: float
     asset_contributions: List[AssetWeight]
     optimization_success: bool = False
+    simulation_id: Optional[str] = None
 
 
 # {
diff --git a/src/calculation_target_reliability/service.py b/src/calculation_target_reliability/service.py
index 6430d0c..efd2059 100644
--- a/src/calculation_target_reliability/service.py
+++ b/src/calculation_target_reliability/service.py
@@ -154,7 +154,7 @@ def project_eaf_improvement(asset: AssetWeight, improvement_factor: float = 0.3)
     projected_eaf = 100 - improved_downtime_pct
     return min(projected_eaf, 99.9)  # Cap at 99.9%
 
-async def identify_worst_eaf_contributors(*, simulation_result, target_eaf: float, db_session: DbSession, oh_session_id: str, collector_db: CollectorDbSession):
+async def identify_worst_eaf_contributors(*, simulation_result, target_eaf: float, db_session: DbSession, oh_session_id: str, collector_db: CollectorDbSession, simulation_id: str):
     """
     Identify equipment that contributes most to plant EAF reduction
 
@@ -205,6 +205,7 @@ async def identify_worst_eaf_contributors(*, simulation_result, target_eaf: floa
         eaf_gap=eaf_gap,
         asset_contributions=selected_asset,
         optimization_success=optimization_success
+        simulation_id=simulation_id
     )
 
 # def optimize_maintenance_priority(*, simulation_result, target_eaf: float):