Minor adjustment to class name