fix the build error in Wangfei's checkin