fixed adding class type