Fix compile

Change-Id: Id7d13635faa4d3848abcdbbb0db097f3591ed4d2