small fix
small fix in controller