X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.scenegraph%2Fsrc%2Forg%2Fsimantics%2Fscenegraph%2Futils%2FNodeUtil.java;h=727b98dbc2295eb499a7aad5fee14d5339db1f87;hb=refs%2Fchanges%2F83%2F183%2F2;hp=fb2f485f6b5935cc9166a59d9728fbaafd244458;hpb=1f4d3bbd7b3180c5702770ce625ccb65a2454f9a;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/utils/NodeUtil.java b/bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/utils/NodeUtil.java
index fb2f485f6..727b98dbc 100644
--- a/bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/utils/NodeUtil.java
+++ b/bundles/org.simantics.scenegraph/src/org/simantics/scenegraph/utils/NodeUtil.java
@@ -978,9 +978,14 @@ public final class NodeUtil {
         if(node instanceof ConnectionNode) {
             return getLocalBounds(node);
         } else if(node instanceof SingleElementNode) {
+            // For normal symbols
             INode image = NodeUtil.findChildByPrefix((SingleElementNode)node, "composite_image");
             if (image == null)
+                // For generic text nodes
                 image = NodeUtil.findChildByPrefix((SingleElementNode) node, "text");
+            if (image == null)
+                // For I/O table diagram flags (value of org.simantics.diagram.flag.FlagSceneGraph.VISUAL_ROOT)
+                image = NodeUtil.findChildByPrefix((SingleElementNode) node, "visual");
             if (image == null)
                 image = NodeUtil.getNearestChildByClass((SingleElementNode) node, FlagNode.class);
             if (image != null)