Fix another test case