From 2ee44df6e298f668b868ed2d010c279c1d219940 Mon Sep 17 00:00:00 2001
From: oseer williams <oseer.williams@gmail.com>
Date: Mon, 6 Nov 2023 12:54:41 -0500
Subject: [PATCH] fix(components): tree list item js error (#2071)

---
 libs/components/src/tree-list/tree-list-item.ts | 4 +++-
 libs/components/src/tree-list/tree-list.ts      | 8 ++++----
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/libs/components/src/tree-list/tree-list-item.ts b/libs/components/src/tree-list/tree-list-item.ts
index e7212a304b..81d2e50b65 100644
--- a/libs/components/src/tree-list/tree-list-item.ts
+++ b/libs/components/src/tree-list/tree-list-item.ts
@@ -3,6 +3,8 @@ import { customElement, property } from 'lit/decorators.js';
 import { classMap } from 'lit/directives/class-map.js';
 import styles from './tree-list-item.scss?inline';
 
+import '../icon/icon';
+
 declare global {
   interface HTMLElementTagNameMap {
     'cv-tree-list-item': CovalentTreeListItem;
@@ -82,4 +84,4 @@ export class CovalentTreeListItem extends LitElement {
   }
 }
 
-export default CovalentTreeListItem;
\ No newline at end of file
+export default CovalentTreeListItem;
diff --git a/libs/components/src/tree-list/tree-list.ts b/libs/components/src/tree-list/tree-list.ts
index c41a2a5fd0..399e61244e 100644
--- a/libs/components/src/tree-list/tree-list.ts
+++ b/libs/components/src/tree-list/tree-list.ts
@@ -32,22 +32,22 @@ document.addEventListener('select', (e: Event) => {
 const handleSelect = (e: Event): void => {
   // All cv-tree-list-item components.
   const items: any[] = Array.from(
-    document.querySelectorAll('cv-tree-list-item')
+    document?.querySelectorAll('cv-tree-list-item')
   );
 
   // Currently selected item.
   const target: HTMLElement = e.target as HTMLElement;
-  const current = target.shadowRoot!.querySelector('div.itemContent');
+  const current = target.shadowRoot?.querySelector('div.itemContent');
 
   // Find previously selected element and remove the styling.
   items.forEach((item) => {
     item
-      .shadowRoot!.querySelector('div.itemContent')
+      .shadowRoot?.querySelector('div.itemContent')
       ?.classList.remove('selected');
   });
 
   // Add the selected styling to the currently selected item.
-  current!.classList.add('selected');
+  current?.classList.add('selected');
 };
 
 export default CovalentTreeList;
\ No newline at end of file