diff --git a/test/docs/each_test.rb b/test/docs/each_test.rb
index 6b825e1..c0ba1a7 100644
--- a/test/docs/each_test.rb
+++ b/test/docs/each_test.rb
@@ -740,8 +740,11 @@ def self.block
 
     stack, signal, (ctx, _) = Trailblazer::Developer::Trace.invoke(activity, [ctx, {}])
 
-    assert_equal Trailblazer::Developer::Trace::Present.(stack,
-      node_options: {stack.to_a[0] => {label: "<a-Each-b>"}}), %{<a-Each-b>
+    output = Trailblazer::Developer::Trace::Present.(stack) do |trace_nodes:, **|
+      {node_options: {trace_nodes[0] => {label: "<a-Each-b>"}}}
+    end
+
+    assert_equal output, %{<a-Each-b>
 |-- Start.default
 |-- a
 |-- Each/1
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 040c267..ef8464f 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -27,7 +27,12 @@ def rehash(ctx, seq:, rehash_raise: false, **)
 Minitest::Spec.class_eval do
   def trace(activity, ctx)
     stack, signal, (ctx, _) = Trailblazer::Developer::Trace.invoke(activity, [ctx, {}])
-    return Trailblazer::Developer::Trace::Present.(stack, node_options: {stack.to_a[0]=>{label: "TOP"}}).gsub(/:\d+/, ""), signal, ctx
+
+    output = Trailblazer::Developer::Trace::Present.(stack) do |trace_nodes:, **|
+      {node_options: {trace_nodes[0] => {label: "TOP"}}}
+    end.gsub(/:\d+/, "")
+
+    return output, signal, ctx
   end
 end
 
diff --git a/trailblazer-macro.gemspec b/trailblazer-macro.gemspec
index eea559f..a04fc06 100644
--- a/trailblazer-macro.gemspec
+++ b/trailblazer-macro.gemspec
@@ -20,7 +20,7 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "minitest"
   spec.add_development_dependency "rake"
   spec.add_development_dependency "trailblazer-developer", ">= 0.1.0", "< 0.2.0"
-  spec.add_dependency "trailblazer-operation", ">= 0.10.0" # TODO: this dependency will be removed. currently needed for tests and for Guard::Result
+  spec.add_dependency "trailblazer-operation", ">= 0.10.1" # TODO: this dependency will be removed. currently needed for tests and for Guard::Result
 
   spec.add_dependency "trailblazer-activity-dsl-linear", ">= 1.2.0", "< 1.3.0"