Fixed test class