Updated the test case