Fix build

Change-Id: I812c96634a22bd3ce220ba74d446daabe9c6724a