Fixed test coverage