diff --git a/packages/clay-core/src/tree-view/TreeViewItem.tsx b/packages/clay-core/src/tree-view/TreeViewItem.tsx
index a615f95806..51fb050065 100644
--- a/packages/clay-core/src/tree-view/TreeViewItem.tsx
+++ b/packages/clay-core/src/tree-view/TreeViewItem.tsx
@@ -401,7 +401,7 @@ export function TreeViewItemStack({
 
 	const item = useItem();
 
-	const childrenArray = React.Children.toArray(children);
+	const childrenCount = React.Children.count(children);
 
 	const nestedChildren =
 		nestedKey && (item[nestedKey] as Array<Record<string, any>>);
@@ -503,9 +503,7 @@ export function TreeViewItemStack({
 				}
 
 				return (
-					<Layout.ContentCol
-						expand={index === childrenArray.length - 1}
-					>
+					<Layout.ContentCol expand={index === childrenCount - 1}>
 						{content}
 					</Layout.ContentCol>
 				);