Add missing class attribute