Minor fix test case