From 29d59bbfb7061c6cbf9e49ce5e70e02748e4a832 Mon Sep 17 00:00:00 2001
From: Michael Harbarth <michael.harbarth@deutschebahn.com>
Date: Wed, 13 Nov 2024 07:45:13 +0100
Subject: [PATCH] fix: associations of subclasses were mapped to the root class
 but should be mapped to the subclass

---
 capellambse_context_diagrams/collectors/tree_view.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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,