fix build

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