Update migration.sh