diff --git a/qt_ui/windows/groundobject/QGroundObjectMenu.py b/qt_ui/windows/groundobject/QGroundObjectMenu.py index 9df4b454..23c6e3ba 100644 --- a/qt_ui/windows/groundobject/QGroundObjectMenu.py +++ b/qt_ui/windows/groundobject/QGroundObjectMenu.py @@ -340,11 +340,7 @@ class QBuyGroupForGroundObjectDialog(QDialog): buy_ewr_layout.addLayout(stretch, 2, 0) # Armored units - - armored_units = db.find_unittype( - PinpointStrike, faction.name - ) # Todo : refactor this legacy nonsense - for unit in set(armored_units): + for unit in set(faction.ground_units): self.buyArmorCombo.addItem( db.unit_type_name_2(unit) + " [$" + str(db.PRICES[unit]) + "M]", userData=unit,