Fix a missing class rename