Updated to new class name, fixes #802