Fix error with class name
Merge pull request #37 from cdjackson/master

Fix error with class name