add test coverage