Fixed a missing class definition.