2 files changed, 8 insertions, 0 deletions
diff --git a/src/ui/include/ui_fsm.hpp b/src/ui/include/ui_fsm.hpp
index 8b80f162..d0bb7b2f 100644
--- a/src/ui/include/ui_fsm.hpp
+++ b/src/ui/include/ui_fsm.hpp
@@ -35,6 +35,8 @@ class UiState : public tinyfsm::Fsm<UiState> {
   /* Fallback event handler. Does nothing. */
   void react(const tinyfsm::Event& ev) {}
 
+  virtual void react(const system_fsm::KeyLockChanged&){};
+
   virtual void react(const system_fsm::DisplayReady&) {}
   virtual void react(const system_fsm::BootComplete&) {}
 
@@ -58,6 +60,8 @@ class Splash : public UiState {
 
 class Interactive : public UiState {
   void entry() override;
+
+  void react(const system_fsm::KeyLockChanged&) override;
 };
 
 class FatalError : public UiState {};
diff --git a/src/ui/ui_fsm.cpp b/src/ui/ui_fsm.cpp
index bdc1f89f..c2ac6d8b 100644
--- a/src/ui/ui_fsm.cpp
+++ b/src/ui/ui_fsm.cpp
@@ -68,6 +68,10 @@ void Interactive::entry() {
   sCurrentScreen.reset(new screens::Menu());
 }
 
+void Interactive::react(const system_fsm::KeyLockChanged& ev) {
+  sDisplay->SetDisplayOn(ev.falling);
+}
+
 }  // namespace states
 }  // namespace ui