Adding coverage back in