Add coverage
Fix extremely minor type