Actual fix for update script. Oops.