Fix className assignment