Update the migration