fix unit test problem.