async engine fix

PrefectHQ · Nov 11, 2024 · f3ae6b3 · f3ae6b3
1 parent ec85769
commit f3ae6b3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/src/prefect/task_engine.py b/src/prefect/task_engine.py
@@ -1344,7 +1344,7 @@ async def start(
         async with self.initialize_run(
             task_run_id=task_run_id, dependencies=dependencies
         ):
-            with trace.use_span(self.span):
+            with trace.use_span(self._span):
                 await self.begin_run()
                 try:
                     yield

diff --git a/tests/test_instrumentation.py b/tests/test_instrumentation.py