fixed error in one of the test cases