Fix compile error due to class rename.