Fixed test properly