fix unit test problem