From a4f2f96f1103e6db96346fdc2604ebc40da4381a Mon Sep 17 00:00:00 2001
From: Predrag Klepic <v-prklepic@microsoft.com>
Date: Tue, 11 Jul 2023 11:39:03 +0200
Subject: [PATCH] Styling fixed in SampleDataTree

---
 src/Explorer/Tree/SampleDataTree.tsx | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/Explorer/Tree/SampleDataTree.tsx b/src/Explorer/Tree/SampleDataTree.tsx
index 0036164c5..f343184fb 100644
--- a/src/Explorer/Tree/SampleDataTree.tsx
+++ b/src/Explorer/Tree/SampleDataTree.tsx
@@ -1,6 +1,7 @@
 import { useCommandBar } from "Explorer/Menus/CommandBar/CommandBarComponentAdapter";
 import TabsBase from "Explorer/Tabs/TabsBase";
 import { useSelectedNode } from "Explorer/useSelectedNode";
+import { getItemName } from "Utils/APITypeUtils";
 import { useTabs } from "hooks/useTabs";
 import React, { useEffect, useState } from "react";
 import CosmosDBIcon from "../../../images/Azure-Cosmos-DB.svg";
@@ -28,7 +29,7 @@ export const SampleDataTree = ({
             label: sampleDataResourceTokenCollection.id(),
             iconSrc: CollectionIcon,
             isExpanded: false,
-            className: "dataResourceTree",
+            className: "sampleCollectionHeader",
             contextMenu: ResourceTreeContextMenuButtonFactory.createSampleCollectionContextMenuButton(),
             onClick: () => {
               // Rewritten version of expandCollapseCollection
@@ -50,7 +51,9 @@ export const SampleDataTree = ({
             onContextMenuOpen: () => useSelectedNode.getState().setSelectedNode(sampleDataResourceTokenCollection),
             children: [
               {
-                label: "Items",
+                label: getItemName(),
+                id: sampleDataResourceTokenCollection.isSampleCollection ? "sampleItems" : "",
+                contextMenu: ResourceTreeContextMenuButtonFactory.createSampleCollectionContextMenuButton(),
               },
             ],
           },