diff --git a/src/lib/palette/palette.component.ts b/src/lib/palette/palette.component.ts
index c250bbe..be8b6a6 100644
--- a/src/lib/palette/palette.component.ts
+++ b/src/lib/palette/palette.component.ts
@@ -356,9 +356,10 @@ export class Palette implements OnInit, OnDestroy, OnChanges {
 
     // Layout palette entry nodes
     paletteNodes.forEach(pnode => {
+      const elementSize = this.palette.findViewByModel(pnode).getBBox();
       let dimension: dia.Size = {
-        width: pnode.get('size').width,
-        height: pnode.get('size').height
+        width: elementSize.width,
+        height: elementSize.height
       };
       if (pnode.get('header')) { //attributes.attrs.header) {
         // Palette entry header