try to fix unit test