diff --git a/packages/ui-stencil-vue/lib/components.ts b/packages/ui-stencil-vue/lib/components.ts
index 38fb382d..24f15988 100644
--- a/packages/ui-stencil-vue/lib/components.ts
+++ b/packages/ui-stencil-vue/lib/components.ts
@@ -61,7 +61,9 @@ export const OramaTextarea = /*@__PURE__*/ defineContainer<JSX.OramaTextarea>('o
 ]);
 
 
-export const OramaToggler = /*@__PURE__*/ defineContainer<JSX.OramaToggler>('orama-toggler', undefined);
+export const OramaToggler = /*@__PURE__*/ defineContainer<JSX.OramaToggler>('orama-toggler', undefined, [
+  'performInitialAnimation'
+]);
 
 
 export const SearchBox = /*@__PURE__*/ defineContainer<JSX.SearchBox>('search-box', undefined, [
diff --git a/packages/ui-stencil/src/components.d.ts b/packages/ui-stencil/src/components.d.ts
index 96bea007..d56e7b23 100644
--- a/packages/ui-stencil/src/components.d.ts
+++ b/packages/ui-stencil/src/components.d.ts
@@ -61,6 +61,7 @@ export namespace Components {
         "value": string | null;
     }
     interface OramaToggler {
+        "performInitialAnimation": boolean;
     }
     interface SearchBox {
         "color": 'dark' | 'light' | 'system';
@@ -222,6 +223,7 @@ declare namespace LocalJSX {
         "value"?: string | null;
     }
     interface OramaToggler {
+        "performInitialAnimation"?: boolean;
     }
     interface SearchBox {
         "color"?: 'dark' | 'light' | 'system';
diff --git a/packages/ui-stencil/src/components/search-box/search-box.tsx b/packages/ui-stencil/src/components/search-box/search-box.tsx
index 76c7bfd8..0816076e 100644
--- a/packages/ui-stencil/src/components/search-box/search-box.tsx
+++ b/packages/ui-stencil/src/components/search-box/search-box.tsx
@@ -41,8 +41,9 @@ export class SearchBox {
         >
           <orama-toggler />
         </div>
-        {/* TODO: How to lazily load components chat component as it won't necessarily be used */}
-        {globalContext.selectedTab === 'search' ? <orama-search /> : <orama-chat />}
+
+        <orama-search style={{ display: globalContext.selectedTab === 'search' ? 'flex' : 'none' }} />
+        <orama-chat style={{ display: globalContext.selectedTab === 'chat' ? 'flex' : 'none' }} />
       </Host>
     )
   }