Fixed an issue with one of the test cases.