diff --git a/telebot/asyncio_handler_backends.py b/telebot/asyncio_handler_backends.py
index e5e2e4f..4ef4555 100644
--- a/telebot/asyncio_handler_backends.py
+++ b/telebot/asyncio_handler_backends.py
@@ -29,6 +29,7 @@ class StatesGroup:
             if not name.startswith('__') and not callable(value) and isinstance(value, State):
                 # change value of that variable
                 value.name = ':'.join((cls.__name__, name))
+                value.group = cls
 
 
 class SkipHandler:
diff --git a/telebot/handler_backends.py b/telebot/handler_backends.py
index b838231..4304317 100644
--- a/telebot/handler_backends.py
+++ b/telebot/handler_backends.py
@@ -163,6 +163,10 @@ class StatesGroup:
             if not name.startswith('__') and not callable(value) and isinstance(value, State):
                 # change value of that variable
                 value.name = ':'.join((cls.__name__, name))
+                value.group = cls
+
+
+
 
     
 class BaseMiddleware: