Fix className generation