From 196706bb29654f92e625d1b26315de0755dffa33 Mon Sep 17 00:00:00 2001
From: Emir Pasic <emirpasic@hotmail.com>
Date: Wed, 7 Sep 2016 03:19:20 +0200
Subject: [PATCH] - set key to key if direct match when inserting into
 red-black tree. fixes the issue when the key is a struct type, which is how
 treeset uses the red-black tree #31

---
 trees/redblacktree/redblacktree.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/trees/redblacktree/redblacktree.go b/trees/redblacktree/redblacktree.go
index a8a496e..f9c9bc9 100644
--- a/trees/redblacktree/redblacktree.go
+++ b/trees/redblacktree/redblacktree.go
@@ -72,6 +72,7 @@ func (tree *Tree) Put(key interface{}, value interface{}) {
 			compare := tree.Comparator(key, node.Key)
 			switch {
 			case compare == 0:
+				node.Key = key
 				node.Value = value
 				return
 			case compare < 0: