diff --git a/ui/loadout_manager.py b/ui/loadout_manager.py
index 058dc04..e136cc4 100644
--- a/ui/loadout_manager.py
+++ b/ui/loadout_manager.py
@@ -1616,19 +1616,19 @@ class LoadoutManagerDialog(QDialog):
         
         # Map slot names to armor slot widgets
         slot_mapping = {
-            'head': self.armor_slots[ArmorSlot.HEAD],
-            'torso': self.armor_slots[ArmorSlot.TORSO],
-            'harness': self.armor_slots[ArmorSlot.TORSO],
-            'arms': self.armor_slots[ArmorSlot.ARMS],
-            'arm guards': self.armor_slots[ArmorSlot.ARMS],
-            'hands': self.armor_slots[ArmorSlot.HANDS],
-            'gloves': self.armor_slots[ArmorSlot.HANDS],
-            'legs': self.armor_slots[ArmorSlot.LEGS],
-            'thigh guards': self.armor_slots[ArmorSlot.LEGS],
-            'shins': self.armor_slots[ArmorSlot.SHINS],
-            'shin guards': self.armor_slots[ArmorSlot.SHINS],
-            'feet': self.armor_slots[ArmorSlot.FEET],
-            'foot guards': self.armor_slots[ArmorSlot.FEET],
+            'head': self.slot_widgets[ArmorSlot.HEAD],
+            'torso': self.slot_widgets[ArmorSlot.TORSO],
+            'harness': self.slot_widgets[ArmorSlot.TORSO],
+            'arms': self.slot_widgets[ArmorSlot.ARMS],
+            'arm guards': self.slot_widgets[ArmorSlot.ARMS],
+            'hands': self.slot_widgets[ArmorSlot.HANDS],
+            'gloves': self.slot_widgets[ArmorSlot.HANDS],
+            'legs': self.slot_widgets[ArmorSlot.LEGS],
+            'thigh guards': self.slot_widgets[ArmorSlot.LEGS],
+            'shins': self.slot_widgets[ArmorSlot.SHINS],
+            'shin guards': self.slot_widgets[ArmorSlot.SHINS],
+            'feet': self.slot_widgets[ArmorSlot.FEET],
+            'foot guards': self.slot_widgets[ArmorSlot.FEET],
         }
         
         pieces_found = 0