diff --git a/game/pretense/pretensemissiongenerator.py b/game/pretense/pretensemissiongenerator.py index a6dc8013..7b18b35b 100644 --- a/game/pretense/pretensemissiongenerator.py +++ b/game/pretense/pretensemissiongenerator.py @@ -135,9 +135,9 @@ class PretenseMissionGenerator(MissionGenerator): self.mission.coalition["red"].add_country(self.e_country) # Add CJTF factions to the coalitions, if they're not being used in the campaign - if CombinedJointTaskForcesBlue not in {self.p_country, self.e_country}: + if CombinedJointTaskForcesBlue.id not in {self.p_country.id, self.e_country.id}: self.mission.coalition["blue"].add_country(CombinedJointTaskForcesBlue()) - if CombinedJointTaskForcesRed not in {self.p_country, self.e_country}: + if CombinedJointTaskForcesRed.id not in {self.p_country.id, self.e_country.id}: self.mission.coalition["red"].add_country(CombinedJointTaskForcesRed()) belligerents = {self.p_country.id, self.e_country.id}