Update the code for the new proto

Bug: 169042373

Test: m

Change-Id: I0e1a97a7275f63c20c9a30fc02c63b21876ae228
1 file changed