code coverage fixes