danielassis-dev
diff --git a/‎src/ifcexpressparser/rt_implementation.py‎
Lines changed: 4 additions & 2 deletions b/‎src/ifcexpressparser/rt_implementation.py‎
Lines changed: 4 additions & 2 deletions
@@ -88,12 +88,14 @@ def __init__(self, mapping):
         for name, type in mapping.schema.entities.items():
             if type.inverse:
                 for attr in type.inverse.elements:
-                    print (attr)
+                    related_entity = mapping.schema.entities[attr.entity]
+                    related_attrs = [a['name'] for a in mapping.get_assignable_arguments(related_entity, include_derived=True)]
+
                     inverse_implementations.append(templates.inverse_implementation % {
                         'type'         : name,
                         'name'         : attr.name,
                         'related_type' : attr.entity,
-                        'index'        : [i for i, a in enumerate(mapping.schema.entities[attr.entity].attributes) if a.name == attr.attribute][0]
+                        'index'        : related_attrs.index(attr.attribute)
                     })
 
         self.str = templates.rt_implementation % {