FIX coverage
Merge pull request #45 from hayd/coverage_fix

FIX coverage