diff --git a/capellambse_context_diagrams/collectors/tree_view.py b/capellambse_context_diagrams/collectors/tree_view.py
index e44bbee..2273134 100644
--- a/capellambse_context_diagrams/collectors/tree_view.py
+++ b/capellambse_context_diagrams/collectors/tree_view.py
@@ -319,11 +319,11 @@ def get_all_classes(
             for prop in cls.owned_properties:
                 process_property(
                     _PropertyInfo(
-                        root,
+                        cls,
                         prop,
                         partition,
                         classes,
-                        cls,
+                        None,
                         max_partition,
                         super,
                         sub,