Fix build

Change-Id: I94d7640c2ec894543da1387ce0c4247d1930b887