修正 Bug ，优化代码