diff --git a/game/db.py b/game/db.py index d05f3f09..9ce1b577 100644 --- a/game/db.py +++ b/game/db.py @@ -940,8 +940,10 @@ def unit_task(unit: UnitType) -> Task: for task, units in UNIT_BY_TASK.items(): if unit in units: return task - elif unit in SAM_CONVERT: - return SAM_CONVERT[unit] + + if unit in SAM_CONVERT: + return unit_task(SAM_CONVERT[unit]) + assert False