Updates to get lcov coverage