Added coverage back in