From 837a38064209f848771096fb8e4554fe7f15aa33 Mon Sep 17 00:00:00 2001
From: Rostislav Wolny <rosta@mailpoet.com>
Date: Wed, 27 Feb 2019 09:18:18 +0100
Subject: [PATCH] Ensure highlighted block when it is re-rendered during edit

[MAILPOET-1805]
---
 .../newsletter_editor/behaviors/HighlightEditingBehavior.js | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/assets/js/src/newsletter_editor/behaviors/HighlightEditingBehavior.js b/assets/js/src/newsletter_editor/behaviors/HighlightEditingBehavior.js
index 374f2d9054..7042117387 100644
--- a/assets/js/src/newsletter_editor/behaviors/HighlightEditingBehavior.js
+++ b/assets/js/src/newsletter_editor/behaviors/HighlightEditingBehavior.js
@@ -21,4 +21,10 @@ BL.HighlightEditingBehavior = Marionette.Behavior.extend({
     this.view.hideTools();
     this.$el.removeClass('mailpoet_highlight');
   },
+  onDomRefresh: function onDomRefresh() {
+    if (this.view._isBeingEdited) {
+      this.view.showTools();
+      this.$el.addClass('mailpoet_highlight');
+    }
+  },
 });