From 5cf8a7a8a403f620a242397aaf8d2b9c685cd8fb Mon Sep 17 00:00:00 2001
From: Sophie 'Sharky' Schumann <sharky@sharky.pw>
Date: Sun, 26 Jul 2020 21:20:49 +0200
Subject: [PATCH] Fixup `mautrix_telegram.scripts.dbms_migrate` import for
 `RoomState` and `UserProfile` as it changed upstream.

---
 mautrix_telegram/scripts/dbms_migrate/__main__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mautrix_telegram/scripts/dbms_migrate/__main__.py b/mautrix_telegram/scripts/dbms_migrate/__main__.py
index f404c17f..23261c0e 100644
--- a/mautrix_telegram/scripts/dbms_migrate/__main__.py
+++ b/mautrix_telegram/scripts/dbms_migrate/__main__.py
@@ -25,7 +25,7 @@ def log(message, end="\n"):
 
 def connect(to):
     from mautrix.util.db import Base
-    from mautrix.bridge.db import RoomState, UserProfile
+    from mautrix.client.state_store.sqlalchemy import RoomState, UserProfile
     from mautrix_telegram.db import (Portal, Message, UserPortal, User, Contact, Puppet, BotChat,
                                      TelegramFile)