修复bug
make it better