Fix test failure due to API change.