Fix flaky test

uber · Dec 28, 2015 · acad054 · acad054
1 parent ae7082e
commit acad054
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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