Fixed test issue