Fixed test failure