fix test that broke due to API change