Added missing UI class.