diff --git a/src/utils/node.ts b/src/utils/node.ts
index 38f7c33f..7f22b0e3 100644
--- a/src/utils/node.ts
+++ b/src/utils/node.ts
@@ -31,9 +31,9 @@ export const canAddChild = (parent: TreeNode, childType: string): boolean => {
   if (!parent || !parent.nodeType) {
     return false;
   }
-  // if (parent.nodeType === NODE_TYPE.ROOT) {
-  //   return getAllowedChildrenTypes(parent.nodeType).includes(childType);
-  // }
+  if (parent.nodeType === NODE_TYPE.ROOT) {
+    return getAllowedChildrenTypes(parent.nodeType).includes(childType);
+  }
   const parentNodeType = isCategoryNodeType(parent.nodeType) ? NODE_TYPE.CATEGORY : parent.nodeType;
   const childNodeType = isCategoryNodeType(childType) ? NODE_TYPE.CATEGORY : childType;
   return getAllowedChildrenTypes(parentNodeType).includes(childNodeType);