Update to follow API change.

Change-Id: Ie792491356a6c77b5f584b621c61788bb5aed6f9
2 files changed