Fix UI build

Change-Id: I5f7dc2c38c39e53d5a9eb2ccddf945d036f52deb