continue to fix unit test