Fix build

Change-Id: Ie233ae9aa7c004ed42862b3f45cd4159142cf3dc