修复BUG

原因是因为之前删了一行很重要的代码