In the RichText editor, after using the remove button, the styles toolbar should be updated for the newly focused element in the editor. It seems to not work in Firefox while this working in Chrome.

Steps to reproduce:

1. Create a content with a RichText field
2. Fill this richtext field with a paragraph and a heading (in this order)
3. Click on the heading
4. Click on the remove button

The heading is correctly removed but the styles toolbar remains configured for the heading and is not moved correctly for the paragraph.