修复编译问题

Signed-off-by: Meathill <meathill@gmail.com>