add more test code coverage