Fix visual entity update

Use aggregatedEntity instead of rawEntity, #773.
mff-uk · Nov 8, 2024 · c338629 · c338629
1 parent 99caf3c
commit c338629
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/applications/conceptual-model-editor/src/visualization.tsx b/applications/conceptual-model-editor/src/visualization.tsx
@@ -914,7 +914,7 @@ function onChangeVisualEntities(
         if (next !== null) {
             // New or changed entity entity.
             if (isVisualNode(next)) {
-                const entity = entities[next.representedEntity]?.rawEntity ?? null;
+                const entity = entities[next.representedEntity]?.aggregatedEntity ?? null;
 
                 if (!isSemanticModelClass(entity) && !isSemanticModelClassUsage(entity)) {
                     console.error("In visual update semantic entity is not class or class usage.", { entity, visual: next });
@@ -941,7 +941,7 @@ function onChangeVisualEntities(
                 }
 
             } else if (isVisualRelationship(next)) {
-                const entity = entities[next.representedRelationship]?.rawEntity ?? null;
+                const entity = entities[next.representedRelationship]?.aggregatedEntity ?? null;
 
                 const isRelationship =
                     isSemanticModelRelationship(entity) ||
@@ -974,7 +974,7 @@ function onChangeVisualEntities(
                 }
 
             } else if (isVisualProfileRelationship(next)) {
-                const entity = entities[next.entity]?.rawEntity ?? null;
+                const entity = entities[next.entity]?.aggregatedEntity ?? null;
 
                 if (!isSemanticModelClassUsage(entity)) {
                     console.error("In visual update semantic entity is not a profile.", { entity, visual: next });