Updated to new class name, fixes #802

| Q             | A
| ------------- | ---
| Bug fix?      | yes
| New feature?  | no
| BC breaks?    | no
| Deprecations? | no
| Tests pass?   | na
| Fixed tickets | #802
| License       | MIT
| Doc PR        | --

*I'm not sure if this is the right way to do it?*