add more unit test code coverage