fix build2

Signed-off-by: TZ <atian25@qq.com>