fix className bug