Fix broken class name