Fixed the test case