added a missing class