diff --git a/src/main/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistant.java b/src/main/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistant.java
index 5d1eca2..44c2812 100644
--- a/src/main/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistant.java
+++ b/src/main/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistant.java
@@ -186,11 +186,13 @@ private void initializeAvatar() {
       avatar.removeFromParent();
     }
     avatar = avatarProvider.get();
-    avatar.setSizeFull();
     this.getElement().appendChild(avatar.getElement());
     this.addAttachListener(ev -> this.getElement().executeJs("return;")
         .then(ev2 -> this.getElement().executeJs("this.childNodes[1].childNodes[0].childNodes[0].appendChild($0)", avatar.getElement())
-            .then(ev3 -> chatWindow.setTarget(avatar))));
+            .then(ev3 -> {
+              chatWindow.setTarget(avatar);
+              avatar.setSizeFull();
+            })));
   }
 
   /**