Fix old reference to removed class