Fixing missing class name