Updated to new class name, fixes #802
Merge pull request #811 from cmfcmf/patch-2

Updated to new class name, fixes #802