improved test coverage
improved test coverage