Added missing class.