diff --git a/apps/studio/components/layouts/TableEditorLayout/TableEditorMenu.tsx b/apps/studio/components/layouts/TableEditorLayout/TableEditorMenu.tsx
index 6bb4e96e35d..c8b1af55e6d 100644
--- a/apps/studio/components/layouts/TableEditorLayout/TableEditorMenu.tsx
+++ b/apps/studio/components/layouts/TableEditorLayout/TableEditorMenu.tsx
@@ -104,6 +104,7 @@ const TableEditorMenu = () => {
             onSelectSchema={(name: string) => {
               setSearchText('')
               setSelectedSchema(name)
+              snap.setSelectedSchemaName(name)
               router.push(`/project/${project?.ref}/editor`)
             }}
             onSelectCreateSchema={() => snap.onAddSchema()}