diff --git a/core/data/alembic/versions/81e44dd6047a_mai2_buddies_support.py b/core/data/alembic/versions/81e44dd6047a_mai2_buddies_support.py
index 7d660b4..348a955 100644
--- a/core/data/alembic/versions/81e44dd6047a_mai2_buddies_support.py
+++ b/core/data/alembic/versions/81e44dd6047a_mai2_buddies_support.py
@@ -11,7 +11,7 @@ import sqlalchemy as sa
 
 # revision identifiers, used by Alembic.
 revision = "81e44dd6047a"
-down_revision = "6a7e8277763b"
+down_revision = "c143b80bd966"
 branch_labels = None
 depends_on = None
 
diff --git a/core/data/schema/base.py b/core/data/schema/base.py
index 9ff3dd6..9d05212 100644
--- a/core/data/schema/base.py
+++ b/core/data/schema/base.py
@@ -6,6 +6,7 @@ from sqlalchemy.engine.base import Connection
 from sqlalchemy.sql import text, func
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy import MetaData, Table, Column
+from sqlalchemy.engine import Row
 from sqlalchemy.types import Integer, String, TIMESTAMP, JSON, INTEGER, TEXT
 from sqlalchemy.schema import ForeignKey
 from sqlalchemy.dialects.mysql import insert