diff --git a/bemani/data/data.py b/bemani/data/data.py index f926b10..7d96897 100644 --- a/bemani/data/data.py +++ b/bemani/data/data.py @@ -135,7 +135,7 @@ class Data: # See if the DB was already created try: cursor = self.__session.execute(text("SELECT COUNT(version_num) AS count FROM alembic_version")) - return cursor.fetchone()["count"] == 1 + return cursor.mappings().fetchone()["count"] == 1 except ProgrammingError: return False