try fix unit test error