Fixing coverage tests.