Modified test to bring coverage back to 100%