Fix build

Change-Id: I9b2f156c14c4fbd104f55f0995a2517f1389008c