diff --git a/ui/hud_overlay_clean.py b/ui/hud_overlay_clean.py
index dbb7617..73ec8bf 100644
--- a/ui/hud_overlay_clean.py
+++ b/ui/hud_overlay_clean.py
@@ -530,6 +530,8 @@ class HUDOverlay(QWidget):
     
     def start_session(self, weapon: str = "Unknown", armor: str = "None",
                       fap: str = "None", loadout: str = "Default",
+                      weapon_dpp: Decimal = Decimal('0.0'),
+                      weapon_cost_per_hour: Decimal = Decimal('0.0'),
                       cost_per_shot: Decimal = Decimal('0.0'),
                       cost_per_hit: Decimal = Decimal('0.0'),
                       cost_per_heal: Decimal = Decimal('0.0')) -> None: