Fix unit test, not updated for prior API change.