fix class name
update