SC2: using enum for campaign check

EnvyDragon · Jan 21, 2024 · d6f0fa1 · d6f0fa1
1 parent 88ff77f
commit d6f0fa1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/worlds/sc2/ClientGui.py b/worlds/sc2/ClientGui.py
@@ -15,7 +15,7 @@
 
 from worlds.sc2.Client import SC2Context, calc_unfinished_missions, parse_unlock
 from worlds.sc2.MissionTables import lookup_id_to_mission, lookup_name_to_mission, campaign_race_exceptions, \
-    SC2Mission, SC2Race
+    SC2Mission, SC2Race, SC2Campaign
 from worlds.sc2.Locations import LocationType, lookup_location_id_to_type
 from worlds.sc2.Options import LocationInclusion
 from worlds.sc2 import SC2World, get_first_mission
@@ -144,7 +144,7 @@ def build_mission_table(self, dt) -> None:
                         campaign_layout_height = (max_mission_count + 2) * 50
                     multi_campaign_layout_height += campaign_layout_height
                     campaign_layout = CampaignLayout(size_hint_y=None, height=campaign_layout_height)
-                    if campaign.campaign_name != "Global":
+                    if campaign != SC2Campaign.GLOBAL:
                         campaign_layout.add_widget(
                             Label(text=campaign.campaign_name, size_hint_y=None, height=25, outline_width=1)
                         )