Update for latest API.