Minor fix in a unit test