add more test coverage