Minor fix in testing