Fixing broken class name