Fixed test for a changed method