diff --git a/packages/web/src/datagrid/DataGrid.svelte b/packages/web/src/datagrid/DataGrid.svelte
index e27fc658f..5cb876226 100644
--- a/packages/web/src/datagrid/DataGrid.svelte
+++ b/packages/web/src/datagrid/DataGrid.svelte
@@ -246,6 +246,7 @@
         skip={!(showMacros && isProApp())}
         collapsed={!expandMacros}
         data-testid="DataGrid_itemMacros"
+        height="20%"
       >
         <MacroManager {...$$props} {managerSize} />
       </WidgetColumnBarItem>
diff --git a/packages/web/src/tabs/QueryTab.svelte b/packages/web/src/tabs/QueryTab.svelte
index b513c1842..af6b55b38 100644
--- a/packages/web/src/tabs/QueryTab.svelte
+++ b/packages/web/src/tabs/QueryTab.svelte
@@ -800,6 +800,7 @@
           onClose={() => {
             isAiAssistantVisible = false;
           }}
+          name='aiAssistant'
         >
           <WidgetsInnerContainer skipDefineWidth flexContainer>
             <QueryAiAssistant