Update the code due to API change.

bug:2553570
Change-Id: Ifa6b2ebbbd0d988deef25afdd013b5429438e52a
1 file changed