Fix className