Modify unit test case