Add missing @class