Restoring code coverage to 100%