Minor fix in example tests