Fix incorrect class name