- fix unit test