Fix compiling

Change-Id: Ifba985f0b68226d42211212f02e842a5a549c36c