Fix test to match current API.