- fix #85

emirpasic · Sep 20, 2018 · 370f7ab · 370f7ab
1 parent 17852f4
commit 370f7ab
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/trees/redblacktree/redblacktree.go b/trees/redblacktree/redblacktree.go
@@ -64,6 +64,8 @@ func NewWithStringComparator() *Tree {
 func (tree *Tree) Put(key interface{}, value interface{}) {
 	var insertedNode *Node
 	if tree.Root == nil {
+		// Assert key is of comparator's type for initial tree
+		tree.Comparator(key, key)
 		tree.Root = &Node{Key: key, Value: value, color: red}
 		insertedNode = tree.Root
 	} else {