Add a missing class