Minor changes to the test case