improved code coverage