diff --git a/spec/graphql/dataloader_spec.rb b/spec/graphql/dataloader_spec.rb
index e791b2f6e0..5684e4d820 100644
--- a/spec/graphql/dataloader_spec.rb
+++ b/spec/graphql/dataloader_spec.rb
@@ -558,7 +558,7 @@ def spawn_fiber
         result
       end
 
-      def spawn_source_task(parent_task, condition)
+      def spawn_source_task(parent_task, condition, trace)
         result = super
         if result
           update_fiber_counts