bug fix related to class name change