diff --git a/src/utils/VisibleNodesGenerator.ts b/src/utils/VisibleNodesGenerator.ts
index 245a870..76fb255 100644
--- a/src/utils/VisibleNodesGenerator.ts
+++ b/src/utils/VisibleNodesGenerator.ts
@@ -309,6 +309,10 @@ class FlattenLayer extends CompressLayer {
           }
         : null,
     )
+    // collapse all nodes on clearing search key
+    if (!searchParams) {
+      this.expandedNodes.clear()
+    }
   }
 }