[BUILD] Fix compile error