Minor changes to test case