improved code coverage of tests