Fix code coverage?