add class name fix