modify unit test bug