Fixed broken class reference