1 files changed, 13 insertions, 0 deletions
diff --git a/treeDraw.go b/treeDraw.go
index 1f3ab11..52fb416 100644
--- a/treeDraw.go
+++ b/treeDraw.go
@@ -182,6 +182,19 @@ func (tk *guiWidget) drawView() {
 		}
 	}
 
+	switch tk.node.WidgetType {
+	case widget.Button:
+		if tk.node.IsEnabled() {
+			if tk.isWindowDense() && tk.isInGrid() {
+				tk.setColorButtonDense()
+			} else {
+				tk.setColorButton()
+			}
+		} else {
+			tk.setColorDisable()
+		}
+	default:
+	}
 	// if you don't do this here, it will be black & white only
 	if tk.color != nil {
 		tk.v.FrameColor = tk.color.frame