diff --git a/gen/briefinggen.py b/gen/briefinggen.py
index ec2f2170..9ede79f3 100644
--- a/gen/briefinggen.py
+++ b/gen/briefinggen.py
@@ -159,7 +159,7 @@ class MissionInfoGenerator:
     Examples of subtypes include briefing generators, kneeboard generators, etc.
     """
 
-    def __init__(self, mission: Mission, game: Optional['Game'] = None) -> None:
+    def __init__(self, mission: Mission, game: 'Game') -> None:
         self.mission = mission
         self.game = game
         self.awacs: List[AwacsInfo] = []
diff --git a/gen/kneeboard.py b/gen/kneeboard.py
index e22d8e03..85b3166a 100644
--- a/gen/kneeboard.py
+++ b/gen/kneeboard.py
@@ -26,7 +26,7 @@ import datetime
 from collections import defaultdict
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Dict, List, Optional, Tuple
+from typing import Dict, List, Optional, Tuple, TYPE_CHECKING
 
 from PIL import Image, ImageDraw, ImageFont
 from dcs.mission import Mission
@@ -42,7 +42,8 @@ from .flights.flight import FlightWaypoint, FlightWaypointType
 from .radios import RadioFrequency
 from .runways import RunwayData
 
-
+if TYPE_CHECKING:
+    from game import Game
 class KneeboardPageWriter:
     """Creates kneeboard images."""
 
@@ -310,8 +311,8 @@ class BriefingPage(KneeboardPage):
 class KneeboardGenerator(MissionInfoGenerator):
     """Creates kneeboard pages for each client flight in the mission."""
 
-    def __init__(self, mission: Mission) -> None:
-        super().__init__(mission)
+    def __init__(self, mission: Mission, game: 'Game') -> None:
+        super().__init__(mission, game)
 
     def generate(self) -> None:
         """Generates a kneeboard per client flight."""