Fix build

Change-Id: Ie5bda438c604cd0828be91ee58c0c682d36e4d7c