Small changes in the test cases