diff --git a/packages/react-components/react-tree/stories/Layouts/TreeItemLayout.stories.tsx b/packages/react-components/react-tree/stories/Layouts/TreeItemLayout.stories.tsx
index a9fb2b1c39b43d..e4c18236396efa 100644
--- a/packages/react-components/react-tree/stories/Layouts/TreeItemLayout.stories.tsx
+++ b/packages/react-components/react-tree/stories/Layouts/TreeItemLayout.stories.tsx
@@ -1,9 +1,9 @@
import * as React from 'react';
import { Tree, TreeItem, TreeItemLayout } from '@fluentui/react-tree';
import {
- CalendarLtr20Regular,
Flag20Regular,
FluentIconsProps,
+ Image20Regular,
Important16Regular,
LockClosed20Regular,
MoreHorizontal20Regular,
@@ -16,10 +16,10 @@ const iconStyleProps: FluentIconsProps = {
primaryFill: 'red',
};
-const Badges = () => (
+const Badges = ({ isImportant, messageCount }: { isImportant?: boolean; messageCount?: number }) => (
<>
-
-
+ {isImportant && }
+ {messageCount && messageCount > 0 && }
>
);
@@ -32,10 +32,10 @@ const RenderIconsAfter = () => (
const Actions = () => (
<>
- } />
+ } />