diff --git a/src/bot/__init__.py b/src/bot/__init__.py
index c8f2914..97f87b9 100644
--- a/src/bot/__init__.py
+++ b/src/bot/__init__.py
@@ -1,6 +1,6 @@
 import functools
 
-import mariadb
+from sqlalchemy.future import Engine
 import telebot
 
 from . import handlers, utils
@@ -8,7 +8,7 @@ from . import handlers, utils
 __all__ = ["handlers", "utils"]
 
 
-def create_bot(token: str, engine: mariadb.Connection) -> telebot.TeleBot:
+def create_bot(token: str, engine: Engine) -> telebot.TeleBot:
     bot = telebot.TeleBot(token)
     bot.register_message_handler(
         functools.partial(handlers.set_master_password, bot, engine),