Fixed test according to review