Minor update to coverage code