Fix the test case