修改 controller