Fixed obsolete class reference