Trying to add a new line but stays inside

Hello,

Thanks for this plugin, is helping me with an extension I am doing to the editor.
My problem is that, after writing a <h> element, when pressing enter the editor stays inside the <h> element. The <br> is being added in the <h1> element.

Now, I solved THIS problem switching the new line behavior to <p>, but that is causing some problems with a <div> element, instead of adding a <p> inside the div, it closes the div adding a </div>