(bug) Must use the old model in the migration script