fixed the test cases