Fixed an issue with the test case.