Fix up the migration