fix className problem