diff --git a/telebot/__init__.py b/telebot/__init__.py
index f034cca..a7f3694 100644
--- a/telebot/__init__.py
+++ b/telebot/__init__.py
@@ -34,7 +34,7 @@ logger.addHandler(console_output_handler)
 logger.setLevel(logging.ERROR)
 
 from telebot import apihelper, util, types
-from telebot.handler_backends import MemoryHandlerBackend, FileHandlerBackend, BaseMiddleware, CancelUpdate, SkipHandler
+from telebot.handler_backends import MemoryHandlerBackend, FileHandlerBackend, BaseMiddleware, CancelUpdate, SkipHandler, State
 from telebot.custom_filters import SimpleCustomFilter, AdvancedCustomFilter
 
 
@@ -2927,7 +2927,7 @@ class TeleBot:
         
 
 
-    def set_state(self, user_id: int, state: Union[int, str], chat_id: int=None) -> None:
+    def set_state(self, user_id: int, state: Union[int, str, State], chat_id: int=None) -> None:
         """
         Sets a new state of a user.
 
diff --git a/telebot/async_telebot.py b/telebot/async_telebot.py
index dce9dec..9571c04 100644
--- a/telebot/async_telebot.py
+++ b/telebot/async_telebot.py
@@ -14,7 +14,7 @@ import telebot.types
 
 # storages
 from telebot.asyncio_storage import StateMemoryStorage, StatePickleStorage
-from telebot.asyncio_handler_backends import CancelUpdate, SkipHandler
+from telebot.asyncio_handler_backends import CancelUpdate, SkipHandler, State
 
 from inspect import signature
 
@@ -3455,7 +3455,7 @@ class AsyncTeleBot:
         return await asyncio_helper.delete_sticker_from_set(self.token, sticker)
 
 
-    async def set_state(self, user_id: int, state: str, chat_id: int=None):
+    async def set_state(self, user_id: int, state: Union[State, int, str], chat_id: int=None):
         """
         Sets a new state of a user.