Update the migration script a little.