Update to match API change.