Small fix in the test cases