Fix up migration files