fix unit failed unit test