From 39e875c1eab56d8f959940fb9d490080c62f5c0a Mon Sep 17 00:00:00 2001
From: _run <khumogo1@gmail.com>
Date: Sat, 25 Sep 2021 22:49:32 +0500
Subject: [PATCH] Update handler_backends.py

---
 telebot/handler_backends.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/telebot/handler_backends.py b/telebot/handler_backends.py
index 673bb46..a6a2dc6 100644
--- a/telebot/handler_backends.py
+++ b/telebot/handler_backends.py
@@ -156,7 +156,7 @@ class StateMachine:
         """
         if chat_id in self._states:
             
-            self._states[int(chat_id)]['state'] = state
+            self._states[chat_id]['state'] = state
         else:
             self._states[chat_id] = {'state': state,'data': {}}
 
@@ -186,14 +186,14 @@ class State:
         :param chat_id:
         :param new_state: new_state of a user
         """
-        self.obj._states[chat_id]['state'] = new_state
+        self.obj.add_state(chat_id,new_state)
 
     def finish(self, chat_id):
         """
         Finish(delete) state of a user.
         :param chat_id:
         """
-        self.obj._states.pop(chat_id)
+        self.obj.delete_state(chat_id)
 
     def retrieve_data(self, chat_id):
         """