fix test coverage