more fix unit test