correcting coverage code