Fixed test after return value change