fix: issue with giving icon a file (#9)

FlorianWoelki · Oct 21, 2021 · ea65e27 · ea65e27
1 parent a981d12
commit ea65e27
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/util.ts b/src/util.ts
@@ -113,10 +113,14 @@ export const addToDOM = (path: string, iconId: string): void => {
     return;
   }
 
-  const titleNode = node.querySelector('.nav-folder-title-content');
+  let titleNode = node.querySelector('.nav-folder-title-content');
   if (!titleNode) {
-    console.error('element with title not found');
-    return;
+    titleNode = node.querySelector('.nav-file-title-content');
+
+    if (!titleNode) {
+      console.error('element with title not found');
+      return;
+    }
   }
 
   const iconNode = document.createElement('div');