Fixed broken class attribute