Fixed the test cases