Add in missing class.