small fix - adapted to class api changes