diff --git a/src/test/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistantLazyLoadingDemo.java b/src/test/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistantLazyLoadingDemo.java
index 9e09518..10847ac 100644
--- a/src/test/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistantLazyLoadingDemo.java
+++ b/src/test/java/com/flowingcode/vaadin/addons/chatassistant/ChatAssistantLazyLoadingDemo.java
@@ -124,6 +124,7 @@ public class ChatAssistantLazyLoadingDemo extends VerticalLayout {
   
   public ChatAssistantLazyLoadingDemo() {
     ChatAssistant chatAssistant = new ChatAssistant();
+    chatAssistant.setClassName("small");
     Span lazyLoadingData = new Span();
     DataProvider<Message,?> dataProvider = DataProvider.fromCallbacks(query->{
       lazyLoadingData.setText("Loading messages from: " + query.getOffset() + ", with limit: " + query.getLimit());
@@ -170,4 +171,5 @@ public ChatAssistantLazyLoadingDemo() {
 
     add(message, lazyLoadingData, chat, chatAssistant);
   }
+  
 }
diff --git a/src/test/resources/META-INF/frontend/styles/chat-assistant-styles-demo.css b/src/test/resources/META-INF/frontend/styles/chat-assistant-styles-demo.css
index d69ef15..4e1a016 100644
--- a/src/test/resources/META-INF/frontend/styles/chat-assistant-styles-demo.css
+++ b/src/test/resources/META-INF/frontend/styles/chat-assistant-styles-demo.css
@@ -21,3 +21,7 @@
     --chatbot-avatar-img: url(chatbot.png);
     --chatbot-avatar-bg-color: white;
   }
+  
+  chat-bot.small vaadin-message {
+  	font-size: var(--lumo-font-size-xxs);
+  }