more code cleanups

Signed-off-by: Jeff Carr <[email protected]>

1 files changed, 2 insertions, 1 deletions

diff --git a/button.go b/button.go
index 6204efe..ccd7670 100644
--- a/button.go
+++ b/button.go
@@ -3,7 +3,8 @@ package gui
 import "git.wit.org/wit/gui/toolkit"
 
 func (parent *Node) NewButton(name string, custom func()) *Node {
-	newNode := parent.newNode(name, toolkit.Button, custom)
+	newNode := parent.newNode(name, toolkit.Button)
+	newNode.Custom = custom
 
 	a := newAction(newNode, toolkit.Add)
 	sendAction(a)