Fixed test fails