Expect pygame.surface.Surface rather than pygame.Surface for screen

Signed-off-by: Olivier Gayot <olivier.gayot@sigexec.com>

2 files changed, 3 insertions, 3 deletions

diff --git a/pycameltris/screens/InGame.py b/pycameltris/screens/InGame.py
index 05c4b2d..93a1d47 100644
--- a/pycameltris/screens/InGame.py
+++ b/pycameltris/screens/InGame.py
@@ -250,9 +250,9 @@ frames_per_gridcell = [48, 43, 38, 33, 28, 23, 18, 13, 8, 6, 5, 5, 5, 4, 4, 4, 3
 
 
 class InGame(Screen):
-    def __init__(self, players: list[Player], screen: pygame.Surface):
+    def __init__(self, players: list[Player], screen: pygame.surface.Surface):
         self.players: list[Player] = players
-        self.screen: pygame.Surface = screen
+        self.screen: pygame.surface.Surface = screen
         self.event_handler: dict[int, Callable[[pygame.event.Event], None]] = {}
 
         def exit(_) -> NoReturn:
diff --git a/pycameltris/screens/Pause.py b/pycameltris/screens/Pause.py
index 885b7f6..766a45c 100644
--- a/pycameltris/screens/Pause.py
+++ b/pycameltris/screens/Pause.py
@@ -15,7 +15,7 @@ def handle_input_pressed(controller: Controller, event: pygame.event.Event):
 
 
 class Pause(Screen):
-    def __init__(self, screen: pygame.Surface):
+    def __init__(self, screen: pygame.surface.Surface):
         self.event_handler: dict[int, Callable[[pygame.event.Event], None]] = {}
 
         def exit(_) -> NoReturn: