Fixing the document due to class name change.