diff --git a/radio/src/gui/colorlcd/preflight_checks.cpp b/radio/src/gui/colorlcd/preflight_checks.cpp
index 311ec65db..0eb858f11 100644
--- a/radio/src/gui/colorlcd/preflight_checks.cpp
+++ b/radio/src/gui/colorlcd/preflight_checks.cpp
@@ -280,8 +280,7 @@ void SwitchWarnMatrix::onPress(uint8_t btn_id)
       bfSet(g_model.switchWarningState, newstate, 3 * sw, 3);
   SET_DIRTY();
 
-  // TODO: save state in object
-  setTextAndState(sw);
+  setTextAndState(btn_id);
 }
 
 bool SwitchWarnMatrix::isActive(uint8_t btn_id)