fixed the test case