Fix a missing @class declaration