Add missing class attribute…