Small update to match API changes

Change-Id: I1d6f38b80ca7874b158444b368537b35d29e2414
3 files changed