Add coverage properly