Update to reflect recent API change.