Import MigrationData, not MigrationRecord

diff --git a/mediagoblin/db/sql/util.py b/mediagoblin/db/sql/util.py
index 659765384b4f4fa747403425f7fdf1d5942c8014..e81cf845bf39e12c89b7466098eb1d47ca23ba91 100644 (file)
--- a/mediagoblin/db/sql/util.py
+++ b/mediagoblin/db/sql/util.py
@@ -50,10 +50,10 @@ class MigrationManager(object):
         self.printer = printer
 
         # For convenience
-        from mediagoblin.db.sql.models import MigrationRecord
+        from mediagoblin.db.sql.models import MigrationData
 
-        self.migration_model = MigrationRecord
-        self.migration_table = MigrationRecord.__table__
+        self.migration_model = MigrationData
+        self.migration_table = MigrationData.__table__
 
     @property
     def sorted_migrations(self):