Reverted the revert
Merge pull request #124 from reviewninja/MitchK/coverage-fix

Reverted the revert