Fix the build

Change-Id: I2771e45054788b631888e6f778997fbe4daf27b5