diff --git a/python_payload/apps/demo_worms4.py b/python_payload/apps/demo_worms4.py
index dfba5f02059185145b6597c19d0c8dc4aaf491d7..4f2e8766719ee90c9f8c5c4fc072ce7cd39a2c47 100644
--- a/python_payload/apps/demo_worms4.py
+++ b/python_payload/apps/demo_worms4.py
@@ -40,6 +40,7 @@ class AppWorms(application.Application):
 
     def on_enter(self):
         # print("on foreground")
+        super().on_enter()
         self.just_shown = True
 
     def draw(self, ctx):
diff --git a/python_payload/st4m/application.py b/python_payload/st4m/application.py
index dcb44fcadfb13036113c545a558718e5684c972b..a8559472ec876b9b13a6e238ab4c5291ef55a4f4 100644
--- a/python_payload/st4m/application.py
+++ b/python_payload/st4m/application.py
@@ -1,4 +1,5 @@
 from st4m.ui.view import ViewWithInputState, ViewTransitionSwipeRight
+from st4m.input import InputState
 
 
 class Application(ViewWithInputState):