improve the test case.