From eb1376eeefd15f3da69badd0110c1db5525f8c1d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ja=CC=81n=20Mikla=CC=81s=CC=8C?= <neosinner@gmail.com>
Date: Fri, 25 Jan 2019 09:42:04 +0100
Subject: [PATCH] Fix post images detection in editor

MAILPOET-1756
---
 lib/Newsletter/Editor/StructureTransformer.php | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/Newsletter/Editor/StructureTransformer.php b/lib/Newsletter/Editor/StructureTransformer.php
index 907e0b218b..469970ecf4 100644
--- a/lib/Newsletter/Editor/StructureTransformer.php
+++ b/lib/Newsletter/Editor/StructureTransformer.php
@@ -72,7 +72,7 @@ class StructureTransformer {
 
   private function filterOutFiguresWithoutImages(array $items) {
     $items = array_filter($items, function (DomNode $item) {
-      if($item->tag === 'figure' && !$item->query('img')) {
+      if($item->tag === 'figure' && $item->query('img')->count() === 0) {
         return false;
       }
       return true;
@@ -81,7 +81,7 @@ class StructureTransformer {
   }
 
   private function isImageElement(DomNode $item) {
-    return $item->tag === 'img' || (in_array($item->tag, ['a', 'figure'], true) && $item->query('img'));
+    return $item->tag === 'img' || (in_array($item->tag, ['a', 'figure'], true) && $item->query('img')->count() > 0);
   }
 
   private function getImageAlignment(DomNode $image) {