Fix reference to class property