Improved upon Update fix

Author: beeker1121

pqueue.go

@@ -233,6 +233,7 @@ func (pq *PriorityQueue) Update(item *PriorityItem, newValue []byte) error {
 		return ErrOutOfBounds
 	}
 
+	item.Key = pq.generateKey(item.Priority, item.ID)
 	item.Value = newValue
 	return pq.db.Put(item.Key, item.Value, nil)
 }

queue.go

@@ -158,6 +158,7 @@ func (q *Queue) Update(item *Item, newValue []byte) error {
 		return ErrOutOfBounds
 	}
 
+	item.Key = idToKey(item.ID)
 	item.Value = newValue
 	return q.db.Put(item.Key, item.Value, nil)
 }

stack.go

@@ -158,6 +158,7 @@ func (s *Stack) Update(item *Item, newValue []byte) error {
 		return ErrOutOfBounds
 	}
 
+	item.Key = idToKey(item.ID)
 	item.Value = newValue
 	return s.db.Put(item.Key, item.Value, nil)
 }