Update to follow internal API changes.

Change-Id: I5447e251a4b1b75561ae9a9a2174330505a6912b
1 file changed