header text breaks image align!

Hi.
It seems to be the bug in CE compared to old normal editor in WP version 4.

When we make image align we’ll like to align text next to this image.

In old editor that function was working for <p> and <h1>,<h2> etc.
So text align to image could be formatted in any fashion.

In classic editor with image align when I convert one of the aligned text lines to <h2>, this line breaks text aligning and this <h2> is landed under the image.

This is very confusing, I’d reaaly like to give oportunity for my customer to choose formatting of text aligned to image using styles from headers.

For now only way to format aligned text is to put spans in html mode and that is too advanced and uncomfortable for typical editors like site owners.