diff --git a/ui/loadout_manager_simple.py b/ui/loadout_manager_simple.py
index eb55cb7..992b976 100644
--- a/ui/loadout_manager_simple.py
+++ b/ui/loadout_manager_simple.py
@@ -365,12 +365,12 @@ class LoadoutManagerDialog(QDialog):
         """Set weapon and calculate cost."""
         # Calculate cost per shot
         decay_pec = Decimal(str(weapon.decay))
-        ammo = Decimal(str(weapon.ammo))
+        ammo = Decimal(str(weapon.ammo_burn))
         cost_per_shot = (decay_pec / Decimal("100")) + (ammo * Decimal("0.0001"))
         
         # Update UI
         self.weapon_btn.setText(weapon.name[:30])
-        self.weapon_info.setText(f"Damage: {weapon.damage} | Range: {weapon.range}")
+        self.weapon_info.setText(f"Damage: {weapon.damage} | Range: {weapon.range_val}")
         self.weapon_decay_label.setText(f"{decay_pec} PEC")
         self.weapon_ammo_label.setText(f"{ammo}")
         self.weapon_cost_label.setText(f"{cost_per_shot:.4f} PED")