Fixed test bug