Fix className

Forgot to replace this one, don't know how this happened ...