modify unit test