Fix build

Change-Id: I4cf10bee6c9a4dbda10642998ffc210cdcf84ab0