1 files changed, 2 insertions, 10 deletions

diff --git a/src/ws/view.js b/src/ws/view.js
index bdc5149..5b69a24 100644
--- a/src/ws/view.js
+++ b/src/ws/view.js
@@ -396,16 +396,8 @@ function collapse(e)
 	localStorage.setItem(id+"_collapsed", collapsed?"true":"false");
 
 	e.stopPropagation();
-	if(collapsed)
-	{
-		node.element.classList.add('collapsed');
-		node.collapse_button.innerHTML = "▼";
-	}
-	else
-	{
-		node.element.classList.remove('collapsed');
-		node.collapse_button.innerHTML = "▲";
-	}
+
+	node.setCollapsed(collapsed);
 }
 
 function changeState(e)