diff --git a/src/Entity/Device.php b/src/Entity/Device.php
index 905b119a..267a09c7 100644
--- a/src/Entity/Device.php
+++ b/src/Entity/Device.php
@@ -43,12 +43,12 @@ use Doctrine\ORM\Mapping as ORM;
 class Device extends PartsContainingDBElement
 {
     /**
-     * @ORM\OneToMany(targetEntity="Category", mappedBy="parent")
+     * @ORM\OneToMany(targetEntity="Device", mappedBy="parent")
      */
     protected $children;
 
     /**
-     * @ORM\ManyToOne(targetEntity="Category", inversedBy="children")
+     * @ORM\ManyToOne(targetEntity="Device", inversedBy="children")
      * @ORM\JoinColumn(name="parent_id", referencedColumnName="id")
      */
     protected $parent;
diff --git a/src/Entity/Supplier.php b/src/Entity/Supplier.php
index 0042baa8..c829d1b4 100644
--- a/src/Entity/Supplier.php
+++ b/src/Entity/Supplier.php
@@ -43,12 +43,12 @@ use Doctrine\ORM\Mapping as ORM;
 class Supplier extends Company
 {
     /**
-     * @ORM\OneToMany(targetEntity="Category", mappedBy="parent")
+     * @ORM\OneToMany(targetEntity="Supplier", mappedBy="parent")
      */
     protected $children;
 
     /**
-     * @ORM\ManyToOne(targetEntity="Category", inversedBy="children")
+     * @ORM\ManyToOne(targetEntity="Supplier", inversedBy="children")
      * @ORM\JoinColumn(name="parent_id", referencedColumnName="id")
      */
     protected $parent;