Fix multiple className issue