Fixed test failure caused by last change