diff --git a/components/lib/tree/UITreeNode.js b/components/lib/tree/UITreeNode.js
index 2c641292e8..dba0d30b01 100644
--- a/components/lib/tree/UITreeNode.js
+++ b/components/lib/tree/UITreeNode.js
@@ -430,13 +430,13 @@ export const UITreeNode = React.memo((props) => {
     };
 
     const onArrowRight = (event) => {
-        if (isLeaf || expanded) {
+        if (isLeaf) {
             return;
         }
 
         event.currentTarget.tabIndex = -1;
 
-        expand(event, true);
+        expand(event, expanded);
     };
 
     const onArrowLeft = (event) => {