diff --git a/python_payload/apps/demo_cap_touch/__init__.py b/python_payload/apps/demo_cap_touch/__init__.py
index 715d06e4991c917e40dec33e84998a6049b92655..b6416de76984f7aa877218631559f1ebb793d751 100644
--- a/python_payload/apps/demo_cap_touch/__init__.py
+++ b/python_payload/apps/demo_cap_touch/__init__.py
@@ -1,7 +1,8 @@
 from st3m import logging
 from st3m.application import Application, ApplicationContext
-from st3m.goose import List
+from st3m.goose import List, Optional
 from st3m.input import InputState
+from st3m.ui.view import ViewManager
 from ctx import Context
 
 log = logging.Log(__name__, level=logging.INFO)