Fix up code coverage