Added missing attribute class