diff --git a/tests/integration/trace/test_zipkin_trace.py b/tests/integration/trace/test_zipkin_trace.py index 05b02a3e..60a71aae 100644 --- a/tests/integration/trace/test_zipkin_trace.py +++ b/tests/integration/trace/test_zipkin_trace.py @@ -256,5 +256,6 @@ def hello(request): trace=True, ) - assert count[0] == (yield f) + count = yield f + assert count == 3 assert test_trace_hook.invalid_traceflag == 0