Added a missing class.