fix the migration