Add missing class declartion.