fix unit test suite