class ref fix