Fix build

Change-Id: I742a4307135e4c962893eb494888554fd4bd2d18