Fixed test class.