Fixed test after API change.