修改Bug，恢复功能