完善测试类,修改BUG