Fixed to use correct class name
Merge pull request #45 from i906/patch-1

Fixed to use correct class name