diff --git a/frontend/js/viewer.js b/frontend/js/viewer.js
index c638ce9..0bac508 100644
--- a/frontend/js/viewer.js
+++ b/frontend/js/viewer.js
@@ -726,7 +726,7 @@ function attachTreeItemLongPress(itemEl, getMenuData) {
 }
 
 export function getVaultIcon(vaultName, size = 16) {
-  const v = allVaults.find((val) => val.name === vaultName);
+  const v = state.allVaults.find((val) => val.name === vaultName);
   const type = v ? v.type : "VAULT";
 
   if (type === "DIR") {