Minor fix of test.