Fixing expected code coverage.