Fix up compile

Change-Id: Id9bb2b97831892a472c1ff970c4f9c2d93799682