Add missing class name