Small changes in the test case