diff --git a/src/mermaidui.js b/src/mermaidui.js index 16b0c1b8d..9f3f3c2ef 100644 --- a/src/mermaidui.js +++ b/src/mermaidui.js @@ -69,17 +69,16 @@ export default class MermaidUI extends Plugin { command.listenTo( buttonView, 'execute', () => { const mermaidItem = editor.execute( 'insertMermaidCommand' ); const mermaidItemViewElement = editor.editing.mapper.toViewElement( mermaidItem ); - let mermaidItemDomElement; - - if ( mermaidItemViewElement ) { - mermaidItemDomElement = view.domConverter.viewToDom( mermaidItemViewElement, document ); - } view.scrollToTheSelection(); view.focus(); - if ( mermaidItemDomElement ) { - mermaidItemDomElement.querySelector( '.ck-mermaid__editing-view' ).focus(); + if ( mermaidItemViewElement ) { + const mermaidItemDomElement = view.domConverter.viewToDom( mermaidItemViewElement, document ); + + if ( mermaidItemDomElement ) { + mermaidItemDomElement.querySelector( '.ck-mermaid__editing-view' ).focus(); + } } } );