1 files changed, 2 insertions, 4 deletions
diff --git a/sca-java-2.x/trunk/modules/composite-diagram/src/main/java/org/apache/tuscany/sca/diagram/layout/EntityBuilder.java b/sca-java-2.x/trunk/modules/composite-diagram/src/main/java/org/apache/tuscany/sca/diagram/layout/EntityBuilder.java
index 64ac2db409..a936d83871 100755
--- a/sca-java-2.x/trunk/modules/composite-diagram/src/main/java/org/apache/tuscany/sca/diagram/layout/EntityBuilder.java
+++ b/sca-java-2.x/trunk/modules/composite-diagram/src/main/java/org/apache/tuscany/sca/diagram/layout/EntityBuilder.java
@@ -77,7 +77,7 @@ public class EntityBuilder {
 
         addInclusions(docEle);
 
-        composite.setAttributes();
+        composite.build();
 
         return composite;
     }
@@ -345,9 +345,7 @@ public class EntityBuilder {
                 setReferences(nVal, elts[i]);
                 setProperties(nVal, elts[i]);
 
-                elts[i].referenceHeight();
-                elts[i].serviceHeight();
-                elts[i].propertyLength();
+                elts[i].build();
             }
         }