Add coverage when upgrade is complete