fix compile due to class name change