修改编译错误，添加头文件