Minor fix to update script