1 files changed, 3 insertions, 5 deletions
diff --git a/treeAdd.go b/treeAdd.go
index 40b03ba..9c71a91 100644
--- a/treeAdd.go
+++ b/treeAdd.go
@@ -28,7 +28,7 @@ func setFake(n *tree.Node) {
 	}
 }
 
-// func (n *node) addWidget(n *tree.Node) {
+// mostly just sets the colors of things
 func addWidget(n *tree.Node) {
 	if !me.ok {
 		log.Log(INFO, "addWidget() START NOT OKAY")
@@ -36,8 +36,7 @@ func addWidget(n *tree.Node) {
 		log.Log(INFO, "addWidget() START NOT OKAY")
 		waitOK()
 	}
-	var tk *guiWidget
-	tk = n.TK.(*guiWidget)
+	tk := n.TK.(*guiWidget)
 	log.Log(INFO, "setStartWH() w.id =", n.WidgetId, "n.name", n.String())
 	switch n.WidgetType {
 	case widget.Root:
@@ -67,10 +66,9 @@ func addWidget(n *tree.Node) {
 		// tk.color = &colorTab
 		return
 	case widget.Button:
+		tk.setColorButton()
 		if tk.node.IsEnabled() {
-			tk.setColorButton()
 		} else {
-			tk.setColorButton()
 			tk.setColorDisable()
 		}
 		return