fix one issue in controller