diff --git a/packages/web/src/tableeditor/newTable.ts b/packages/web/src/tableeditor/newTable.ts
index c7b4a8c68..8e5dd82f9 100644
--- a/packages/web/src/tableeditor/newTable.ts
+++ b/packages/web/src/tableeditor/newTable.ts
@@ -31,6 +31,7 @@ export default function newTable(connection, database) {
             },
           ],
           primaryKey: {
+            constraintType: 'primaryKey',
             columns: [
               {
                 columnName: 'id',
diff --git a/packages/web/src/tabs/QueryTab.svelte b/packages/web/src/tabs/QueryTab.svelte
index 9113ccd5c..59d05e87b 100644
--- a/packages/web/src/tabs/QueryTab.svelte
+++ b/packages/web/src/tabs/QueryTab.svelte
@@ -149,7 +149,7 @@
   let domEditor;
   let domToolStrip;
   let intervalId;
-  let isAiAssistantVisible = localStorage.getItem(`tabdata_isAiAssistantVisible_${tabid}`) == 'true';
+  let isAiAssistantVisible = isProApp() && localStorage.getItem(`tabdata_isAiAssistantVisible_${tabid}`) == 'true';
 
   onMount(() => {
     intervalId = setInterval(() => {