2 files changed, 10 insertions, 2 deletions
diff --git a/pycameltris/screens/InGame.py b/pycameltris/screens/InGame.py
index aa3e53b..7363f5a 100644
--- a/pycameltris/screens/InGame.py
+++ b/pycameltris/screens/InGame.py
@@ -2,6 +2,7 @@ import contextlib
 from functools import partial
 import random
 import sys
+from typing import NoReturn
 
 import pygame
 
@@ -254,7 +255,10 @@ class InGame(Screen):
         self.screen = screen
         self.event_handler = dict()
 
-        self.event_handler[pygame.QUIT] = lambda _: sys.exit()
+        def exit(_) -> NoReturn:
+            sys.exit()
+
+        self.event_handler[pygame.QUIT] = exit
         self.event_handler[pygame.KEYDOWN] = partial(handle_input_pressed, KeyboardController, self.players)
         self.event_handler[pygame.KEYUP] = partial(handle_input_released, KeyboardController, self.players)
         self.event_handler[pygame.JOYBUTTONDOWN] = partial(handle_input_pressed, JoystickController, self.players)
diff --git a/pycameltris/screens/Pause.py b/pycameltris/screens/Pause.py
index baca5ca..1502be7 100644
--- a/pycameltris/screens/Pause.py
+++ b/pycameltris/screens/Pause.py
@@ -1,5 +1,6 @@
 import contextlib
 import sys
+from typing import NoReturn
 
 import pygame
 
@@ -17,7 +18,10 @@ class Pause(Screen):
     def __init__(self, screen):
         self.event_handler = dict()
 
-        self.event_handler[pygame.QUIT] = lambda _: sys.exit()
+        def exit(_) -> NoReturn:
+            sys.exit()
+
+        self.event_handler[pygame.QUIT] = exit
         self.event_handler[pygame.JOYBUTTONDOWN] = lambda evt: handle_input_pressed(JoystickController(evt.joy), evt)
         self.event_handler[pygame.KEYDOWN] = lambda evt: handle_input_pressed(KeyboardController(pygame.key), evt)