diff --git a/src/migrations/versions/89121d2c96d8_add_sheet_id_to_archive_table.py b/src/migrations/versions/89121d2c96d8_add_sheet_id_to_archive_table.py
index bbdbee1..3011cf6 100644
--- a/src/migrations/versions/89121d2c96d8_add_sheet_id_to_archive_table.py
+++ b/src/migrations/versions/89121d2c96d8_add_sheet_id_to_archive_table.py
@@ -19,7 +19,7 @@ depends_on = None
 
 def upgrade() -> None:
     conn = op.get_bind()
-    inspector = Inspector.from_engine(conn)
+    inspector = sa.inspect(conn)
     columns = [col['name'] for col in inspector.get_columns('archives')]
 
     if 'sheet_id' not in columns:
@@ -30,7 +30,7 @@ def upgrade() -> None:
 
 def downgrade() -> None:
     conn = op.get_bind()
-    inspector = Inspector.from_engine(conn)
+    inspector = sa.inspect(conn)
     foreign_keys = [fk['name'] for fk in inspector.get_foreign_keys('archives')]
     columns = [col['name'] for col in inspector.get_columns('archives')]
 
diff --git a/src/migrations/versions/fa012ec405b8_add_columns_to_groups_table.py b/src/migrations/versions/fa012ec405b8_add_columns_to_groups_table.py
index be94c98..f0577ea 100644
--- a/src/migrations/versions/fa012ec405b8_add_columns_to_groups_table.py
+++ b/src/migrations/versions/fa012ec405b8_add_columns_to_groups_table.py
@@ -19,7 +19,7 @@ depends_on = None
 
 def upgrade() -> None:
     conn = op.get_bind()
-    inspector = Inspector.from_engine(conn)
+    inspector = sa.inspect(conn)
     columns = [col['name'] for col in inspector.get_columns('groups')]
 
     if 'description' not in columns:
@@ -36,7 +36,7 @@ def upgrade() -> None:
 
 def downgrade() -> None:
     conn = op.get_bind()
-    inspector = Inspector.from_engine(conn)
+    inspector = sa.inspect(conn)
     columns = [col['name'] for col in inspector.get_columns('groups')]
 
     column_names = ['description', 'orchestrator', 'orchestrator_sheet', 'permissions', 'domains']