Fixed a missing class