修正主函数的bug