1 files changed, 8 insertions, 8 deletions
diff --git a/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj b/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj
index 83e46be3..bd2a4d47 100644
--- a/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj
+++ b/src/main/aspectj/kamon/instrumentation/PromiseCompletingRunnableInstrumentation.aj
@@ -4,26 +4,26 @@ import kamon.TraceContext;
 import scala.concurrent.impl.Future;
 import scala.Option;
 
-public aspect PromiseCompletingRunnableInstrumentation {
+privileged public aspect PromiseCompletingRunnableInstrumentation {
+
+    declare parents : Future.PromiseCompletingRunnable extends TraceContextHolder;
 
-    public final Option<TraceContext> Future.PromiseCompletingRunnable.context = TraceContext.current();
 
     pointcut run(Future.PromiseCompletingRunnable runnable)
             : execution(* scala.concurrent.impl.Future.PromiseCompletingRunnable.run()) && this(runnable);
 
-    void around(Future.PromiseCompletingRunnable runnable)
+    void around(TraceContextHolder runnable)
         : run(runnable) {
 
-        if(runnable.context.isDefined()) {
-
-            TraceContext.set((TraceContext) runnable.context.get());
+        if(runnable.getContext().isDefined()) {
+            System.out.println("########################################################3 There is some context");
+            TraceContext.set(runnable.getContext().get());
             proceed(runnable);
             TraceContext.clear();
 
         } else {
+            System.out.println("########################################################3 There is NOOOOO context");
             proceed(runnable);
         }
-
-
     }
 }