Corrected one of the test cases