Get code coverage back to 100%