Fix build error

Change-Id: I5c992189dc21683fc9e5244f4824d7c42e4ed9cd