diff --git a/mautrix_telegram/scripts/dbms_migrate/__main__.py b/mautrix_telegram/scripts/dbms_migrate/__main__.py
index bc33bcc9..eed45071 100644
--- a/mautrix_telegram/scripts/dbms_migrate/__main__.py
+++ b/mautrix_telegram/scripts/dbms_migrate/__main__.py
@@ -15,8 +15,8 @@ args = parser.parse_args()
 
 
 def connect(to):
-    import mautrix_telegram.base as base
-    base.Base = declarative_base()
+    import mautrix_telegram.db.base as base
+    base.Base = declarative_base(cls=base.BaseBase)
     from mautrix_telegram.db import (Portal, Message, UserPortal, User, RoomState, UserProfile,
                                      Contact, Puppet, BotChat, TelegramFile)
     db_engine = sql.create_engine(to)
diff --git a/setup.py b/setup.py
index df4c7962..2a5c55f7 100644
--- a/setup.py
+++ b/setup.py
@@ -40,7 +40,7 @@ setuptools.setup(
         "future-fstrings>=0.4.2",
         "python-magic>=0.4.15,<0.5",
         "telethon>=1.5.5,<1.6",
-        "telethon-session-sqlalchemy>=0.2.8,<0.3",
+        "telethon-session-sqlalchemy>=0.2.9,<0.3",
     ],
     extras_require=extras,