fix unit test issue