1 files changed, 18 insertions, 16 deletions
diff --git a/app/assets/javascripts/image_diff/replaced_image_diff.js b/app/assets/javascripts/image_diff/replaced_image_diff.js
index 5c7e4d5e5a6..4abd13fb472 100644
--- a/app/assets/javascripts/image_diff/replaced_image_diff.js
+++ b/app/assets/javascripts/image_diff/replaced_image_diff.js
@@ -70,21 +70,23 @@ export default class ReplacedImageDiff extends ImageDiff {
     // Image_file.js has a fade animation of 200ms for loading the view
     // Need to wait an additional 250ms for the images to be displayed
     // on window in order to re-normalize their dimensions
-    setTimeout(() => {
-      // Generate badge coordinates on new view
-      this.renderBadges();
-
-      // Re-render indicator in new view
-      if (indicator.removed) {
-        const normalizedIndicator = imageDiffHelper
-          .resizeCoordinatesToImageElement(this.imageEl, {
-            x: indicator.x,
-            y: indicator.y,
-            width: indicator.image.width,
-            height: indicator.image.height,
-          });
-        imageDiffHelper.showCommentIndicator(this.imageFrameEl, normalizedIndicator);
-      }
-    }, 250);
+    setTimeout(this.renderNewView.bind(this, indicator), 250);
+  }
+
+  renderNewView(indicator) {
+    // Generate badge coordinates on new view
+    this.renderBadges();
+
+    // Re-render indicator in new view
+    if (indicator.removed) {
+      const normalizedIndicator = imageDiffHelper
+        .resizeCoordinatesToImageElement(this.imageEl, {
+          x: indicator.x,
+          y: indicator.y,
+          width: indicator.image.width,
+          height: indicator.image.height,
+        });
+      imageDiffHelper.showCommentIndicator(this.imageFrameEl, normalizedIndicator);
+    }
   }
 }