Fix obsolete class name