diff --git a/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertColumnsDialog.svelte b/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertColumnsDialog.svelte
index bd4f96f..7a241cc 100644
--- a/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertColumnsDialog.svelte
+++ b/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertColumnsDialog.svelte
@@ -7,6 +7,7 @@
   import {INSERT_LAYOUT_COMMAND} from '../../../core/plugins/ColumnsLayout/LayoutItemNode.js';
   import DropDownItem from '../../generic/dropdown/DropDownItem.svelte';
   import DropDown from '../../generic/dropdown/DropDown.svelte';
+  import {tick} from 'svelte';
 
   const editor = getEditor();
   const activeEditor = getActiveEditor();
@@ -16,8 +17,9 @@
     showModal = true;
   }
 
-  function close() {
+  async function close() {
     showModal = false;
+    await tick();
     getCommands().FocusEditor.execute(editor);
   }
 
diff --git a/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertTableDialog.svelte b/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertTableDialog.svelte
index 640cde8..ee6810d 100644
--- a/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertTableDialog.svelte
+++ b/packages/svelte-lexical/src/lib/components/toolbar/dialogs/InsertTableDialog.svelte
@@ -4,7 +4,7 @@
   import ModalDialog from '../../generic/dialog/ModalDialog.svelte';
   import {getCommands} from '$lib/core/commands.js';
   import NumberInput from '$lib/components/generic/input/NumberInput.svelte';
-
+  import {tick} from 'svelte';
   let rows = '5';
   let columns = '5';
   let isDisabled = true;
@@ -26,8 +26,9 @@
     showModal = true;
   }
 
-  function close() {
+  async function close() {
     showModal = false;
+    await tick();
     getCommands().FocusEditor.execute($activeEditor);
   }