Migration script update