Fix broken class ref