diff --git a/src/ColumnsBlock/EditBlockWrapper.jsx b/src/ColumnsBlock/EditBlockWrapper.jsx
index 8e9ae2e..8d3ef1e 100644
--- a/src/ColumnsBlock/EditBlockWrapper.jsx
+++ b/src/ColumnsBlock/EditBlockWrapper.jsx
@@ -66,7 +66,9 @@ class EditBlockWrapper extends React.Component {
       block,
       data,
       onDeleteBlock,
+      onInsertBlock,
       onMutateBlock,
+      onSelectBlock,
       selected,
     } = blockProps;
     const type = data['@type'];
@@ -139,8 +141,12 @@ class EditBlockWrapper extends React.Component {
               {this.state.addNewBlockOpened && (
                 <BlockChooser
                   onMutateBlock={(id, value) => {
-                    this.setState({ addNewBlockOpened: false });
                     onMutateBlock(id, value);
+                    this.setState({ addNewBlockOpened: false });
+                  }}
+                  onInsertBlock={(id, value) => {
+                    onSelectBlock(onInsertBlock(id, value));
+                    this.setState({ addNewBlockOpened: false });
                   }}
                   currentBlock={block}
                   allowedBlocks={allowedBlocks}