diff --git a/src/app/views/query-response/pivot-items/pivot-items.tsx b/src/app/views/query-response/pivot-items/pivot-items.tsx
index 8eb2ddc8a3..799787c35b 100644
--- a/src/app/views/query-response/pivot-items/pivot-items.tsx
+++ b/src/app/views/query-response/pivot-items/pivot-items.tsx
@@ -53,8 +53,8 @@ export const getPivotItems = () => {
         <Icon iconName={link.itemIcon} style={{ paddingRight: 5 }} />
         {link.headerText}
 
-        {link.ariaLabel === 'Adaptive Cards' && showDotIfAdaptiveCardPresent()}
-        {link.ariaLabel === 'Graph Toolkit' && showDotIfGraphToolkitPresent()}
+        {link.itemKey === 'adaptive-cards' && showDotIfAdaptiveCardPresent()}
+        {link.itemKey === 'toolkit-component' && showDotIfGraphToolkitPresent()}
       </TooltipHost>
     );
   }