diff --git a/src/Form/Field/Tree.php b/src/Form/Field/Tree.php
index e21275ce2..7bf7f5f84 100755
--- a/src/Form/Field/Tree.php
+++ b/src/Form/Field/Tree.php
@@ -81,7 +81,7 @@ public function treeState(bool $value = true)
     {
         $this->options['checkbox']['three_state'] = $value;
 
-        return $this;
+        return $this->exceptParentNode($value);
     }
 
     /**