1 files changed, 4 insertions, 2 deletions
diff --git a/src/main/scala/kamon/TraceContext.scala b/src/main/scala/kamon/TraceContext.scala
index 351446f3..0bfcd74b 100644
--- a/src/main/scala/kamon/TraceContext.scala
+++ b/src/main/scala/kamon/TraceContext.scala
@@ -53,12 +53,14 @@ object ThreadLocalTraceEntryStorage extends TraceEntryStorage {
 
   private val storage = new ThreadLocal[List[TraceEntry]] {
     override def initialValue(): List[TraceEntry] = Nil
-    def update(f: List[TraceEntry] => List[TraceEntry]) = set(f(get()))
   }
 
+  def update(f: List[TraceEntry] => List[TraceEntry]) = storage set f(storage.get)
+
   def store(entry: TraceEntry): Boolean = {
-    storage.update(entry :: _)
+    update(entry :: _)
     true
   }
 }
 
+