minor class name fix