fix the build error

Signed-off-by: kevinhuhui <huhuikevin@gmail.com>