Removed wrong className assignment