From 6f65637f6bb25f363422c3e7ef2060a68d28d026 Mon Sep 17 00:00:00 2001
From: walterroach <37820425+walterroach@users.noreply.github.com>
Date: Fri, 27 Nov 2020 17:44:51 -0600
Subject: [PATCH] Fix bug #353

---
 gen/triggergen.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/gen/triggergen.py b/gen/triggergen.py
index ba87bb3e..eba9f4f4 100644
--- a/gen/triggergen.py
+++ b/gen/triggergen.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from dcs.action import MarkToAll
 from dcs.condition import TimeAfter
 from dcs.mission import Mission
@@ -5,7 +7,7 @@ from dcs.task import Option
 from dcs.translation import String
 from dcs.triggers import Event, TriggerOnce
 from dcs.unit import Skill
-
+from dcs.unitgroup import FlyingGroup
 from .conflictgen import Conflict
 
 PUSH_TRIGGER_SIZE = 3000
@@ -73,8 +75,9 @@ class TriggersGenerator:
                 continue
 
             for country in coalition.countries.values():
-                for plane_group in country.plane_group:
-                    for plane_unit in plane_group.units:
+                flying_groups = country.plane_group + country.helicopter_group  # type: FlyingGroup
+                for flying_group in flying_groups:
+                    for plane_unit in flying_group.units:
                         if plane_unit.skill != Skill.Client and plane_unit.skill != Skill.Player:
                             plane_unit.skill = Skill(skill_level[0])