Fixed test case that broke due to API change.