Fixed update script from migration