Fixed trying to access null object when setting image to null

BitBagCommerce · Jun 28, 2018 · 1cf08d4 · 1cf08d4
1 parent 3655032
commit 1cf08d4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/Entity/BlockTranslation.php b/src/Entity/BlockTranslation.php
@@ -94,7 +94,9 @@ public function getImage(): ?BlockImageInterface
      */
     public function setImage(?BlockImageInterface $image): void
     {
-        $image->setOwner($this);
+        if ($image !== null) {
+            $image->setOwner($this);
+        }
 
         $this->image = $image;
     }

diff --git a/src/Entity/PageTranslation.php b/src/Entity/PageTranslation.php
@@ -104,7 +104,9 @@ public function getImage(): ?PageImageInterface
      */
     public function setImage(?PageImageInterface $image): void
     {
-        $image->setOwner($this);
+        if ($image !== null) {
+            $image->setOwner($this);
+        }
 
         $this->image = $image;
     }