Fixing coverage mode.
Merge pull request #375 from mithro/coverage

Fixing coverage mode.