1 files changed, 11 insertions, 0 deletions
diff --git a/src/displayapp/screens/Alarm.cpp b/src/displayapp/screens/Alarm.cpp
index 879e50d..3ae6480 100644
--- a/src/displayapp/screens/Alarm.cpp
+++ b/src/displayapp/screens/Alarm.cpp
@@ -255,6 +255,17 @@ void Alarm::SetAlerting() {
   lv_obj_set_hidden(btnStop, false);
   taskStopAlarm = lv_task_create(StopAlarmTaskCallback, pdMS_TO_TICKS(60 * 1000), LV_TASK_PRIO_MID, this);
   systemTask.PushMessage(System::Messages::DisableSleeping);
+  SetEnableButtonState();
+}
+
+void Alarm::StopAlarm() {
+  alarmController.StopAlerting();
+  SetEnableButtonState();
+  if (taskStopAlarm != nullptr) {
+    lv_task_del(taskStopAlarm);
+    taskStopAlarm = nullptr;
+  }
+  systemTask.PushMessage(System::Messages::EnableSleeping);
 }
 
 void Alarm::StopAlerting() {