diff --git a/alembic/versions/6ca3d74d51e4_move_state_store_to_main_database.py b/alembic/versions/6ca3d74d51e4_move_state_store_to_main_database.py
index f9002a00..b5f583da 100644
--- a/alembic/versions/6ca3d74d51e4_move_state_store_to_main_database.py
+++ b/alembic/versions/6ca3d74d51e4_move_state_store_to_main_database.py
@@ -12,7 +12,7 @@ import json
 import re
 
 from mautrix_telegram.config import Config
-from mautrix_telegram.base import Base
+from mautrix_telegram.db import Base
 
 # revision identifiers, used by Alembic.
 revision = "6ca3d74d51e4"