Fix classname.
Merge pull request #467 from Hywan/fix_classname

Fix classname