fixing the 'class' atttibute.