Fix className variable
Merge pull request #125 from jonasdt/master

Fix missing variable rename.