修正编译报错

自作聪明的做了一些更改，现在又改回来了。。。