diff --git a/treeAdd.go b/treeAdd.go
index d5e64b5..b56ff7b 100644
--- a/treeAdd.go
+++ b/treeAdd.go
@@ -62,12 +62,16 @@ func addWidget(n *tree.Node) {
 		return
 	case widget.Button:
 		nw.color = &colorButton
+		return
 	case widget.Checkbox:
 		nw.color = &colorCheckbox
+		return
 	case widget.Dropdown:
 		nw.color = &colorDropdown
+		return
 	case widget.Combobox:
 		nw.color = &colorCombobox
+		return
 	case widget.Box:
 		nw.color = &colorBox
 		nw.isFake = true