1 files changed, 4 insertions, 0 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 ed2d8c18c7..485ced717d 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
@@ -64,6 +64,10 @@ public class EntityBuilder {
 
         LayoutBuilder buildLayout = new LayoutBuilder(comps, conns, Constant.MAX_LEVELS);
         buildLayout.placeEntities();
+        composite.setHeight(buildLayout.getTotalHeight());
+        composite.setWidth(buildLayout.getTotalWidth());
+        composite.setMaxInternalLane(buildLayout.getTotalLane());
+        composite.setMaxInternalLevel(buildLayout.getTotalLevel());
 
         //System.out.println("conns "+conns[0][0]);