little fix in controller