[Wen Hao] - fix build.