fix test unit problem.