fixed broken class