修改test类