Fixed a missing classes