代码优化
fixed #644