From 7658b1ed9a6fc59b2aa8d902beae587b3453f8ea Mon Sep 17 00:00:00 2001
From: Evgenii Meshcheriakov <evgeny.meshcheryakov@gmail.com>
Date: Sun, 16 Oct 2022 16:08:17 +0300
Subject: [PATCH] statehandler: fix #20

---
 StateHandler/src/StateHandler.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/StateHandler/src/StateHandler.cpp b/StateHandler/src/StateHandler.cpp
index f706c03..e56c2c8 100644
--- a/StateHandler/src/StateHandler.cpp
+++ b/StateHandler/src/StateHandler.cpp
@@ -102,6 +102,7 @@ StateHandler::stateManual (const Event &event)
   switch (event.type)
     {
     case Event::eEnter:
+      displaySet (saved_set_value[MANUAL], saved_curr_value[MANUAL]);
       break;
     case Event::eExit:
       _lcd->clear ();
@@ -121,6 +122,7 @@ StateHandler::stateAuto (const Event &event)
   switch (event.type)
     {
     case Event::eEnter:
+      displaySet (saved_set_value[AUTO], saved_curr_value[AUTO]);
       break;
     case Event::eExit:
       _lcd->clear ();
@@ -162,6 +164,6 @@ StateHandler::save (int eventValue, int counterValue, size_t mode)
     {
       saved_curr_value[mode] = eventValue;
       saved_set_value[mode] = counterValue;
-      displaySet (counterValue, eventValue);
+      displaySet (saved_set_value[mode], saved_curr_value[mode]);
     }
 }