Fix update script