fix problem in controller