adding missing class attribute