diff --git a/src/Oro/Bundle/CatalogBundle/Entity/Category.php b/src/Oro/Bundle/CatalogBundle/Entity/Category.php
index 09d43cefaa3..dcf9de576df 100644
--- a/src/Oro/Bundle/CatalogBundle/Entity/Category.php
+++ b/src/Oro/Bundle/CatalogBundle/Entity/Category.php
@@ -339,7 +339,7 @@ class Category extends ExtendCategory implements SluggableInterface, DatesAwareI
     /**
      * @var integer
      *
-     * @Gedmo\TreeRoot
+     * @Gedmo\TreeRoot(identifierMethod="getRoot")
      * @ORM\Column(name="tree_root", type="integer", nullable=true)
      * @ConfigField(
      *      defaultValues={
diff --git a/src/Oro/Component/Tree/Entity/TreeTrait.php b/src/Oro/Component/Tree/Entity/TreeTrait.php
index 2197e3b6ed6..b5d31067b53 100644
--- a/src/Oro/Component/Tree/Entity/TreeTrait.php
+++ b/src/Oro/Component/Tree/Entity/TreeTrait.php
@@ -34,7 +34,7 @@ trait TreeTrait
     /**
      * @var integer
      *
-     * @Gedmo\TreeRoot
+     * @Gedmo\TreeRoot(identifierMethod="getRoot")
      * @ORM\Column(name="tree_root", type="integer", nullable=true)
      */
     protected $root;