added missing line in ui controller