Fix build

Change-Id: I2e951b1b059f40468eb4a8435eb537c937dbd772