fixed reference to class