diff --git a/src/swgoh_comlink/helpers/_territory_battle.py b/src/swgoh_comlink/helpers/_territory_battle.py index a377426..de3b1c5 100644 --- a/src/swgoh_comlink/helpers/_territory_battle.py +++ b/src/swgoh_comlink/helpers/_territory_battle.py @@ -520,11 +520,11 @@ def get_tb_mission_scores( definitions = _select_definitions(territory_battle_definitions, None, tb_id, func_name) _check_collection("tables", tables, func_name) - wanted = { - zone.get("encounterRewardTableId") + wanted: set[str] = { + table_id for definition in definitions for _, zone in _mission_zones(definition) - if zone.get("encounterRewardTableId") + if (table_id := zone.get("encounterRewardTableId")) } # 'table' holds every kind of game table; only parse the ones a mission names. curves: dict[str, list[int]] = {}