fix up test case