minor changes to test case