diff --git a/src/ifccsv/ifccsv.py b/src/ifccsv/ifccsv.py
index 8fc4adb5a1..e415f4bad2 100755
--- a/src/ifccsv/ifccsv.py
+++ b/src/ifccsv/ifccsv.py
@@ -45,7 +45,7 @@ except:
 class IfcAttributeSetter:
     @staticmethod
     def set_element_key(ifc_file, element, key, value):
-        if key == "type" and element.is_a() != value:
+        if key == "class" and element.is_a() != value:
             return ifcopenshell.util.schema.reassign_class(ifc_file, element, value)
         if hasattr(element, key):
             setattr(element, key, value)