Updated to API change.