有bug，修复一下