diff --git a/core/data/database.py b/core/data/database.py
index bed3e79..fb36ebd 100644
--- a/core/data/database.py
+++ b/core/data/database.py
@@ -25,7 +25,7 @@ from core.utils import MISSING, Utils
 
 class Data:
     engine: ClassVar[AsyncEngine] = MISSING
-    session: ClassVar[sessionmaker[AsyncSession]] = MISSING
+    session: ClassVar["sessionmaker[AsyncSession]"] = MISSING
     user: ClassVar[UserData] = MISSING
     arcade: ClassVar[ArcadeData] = MISSING
     card: ClassVar[CardData] = MISSING
diff --git a/core/data/schema/base.py b/core/data/schema/base.py
index adacb9c..80ab74b 100644
--- a/core/data/schema/base.py
+++ b/core/data/schema/base.py
@@ -39,7 +39,7 @@ event_log: Table = Table(
 
 
 class BaseData:
-    def __init__(self, cfg: CoreConfig, conn: sessionmaker[AsyncSession]) -> None:
+    def __init__(self, cfg: CoreConfig, conn: "sessionmaker[AsyncSession]") -> None:
         self.config = cfg
         self.conn = conn
         self.logger = logging.getLogger("database")