调整代码 & fix bug