Fix error with update script