diff --git a/python/ray/workflow/tests/test_basic_workflows_2.py b/python/ray/workflow/tests/test_basic_workflows_2.py
index acecfb14dc01..afd97b3a4167 100644
--- a/python/ray/workflow/tests/test_basic_workflows_2.py
+++ b/python/ray/workflow/tests/test_basic_workflows_2.py
@@ -174,7 +174,7 @@ def wait(obj_ref):
     except Exception:
         v = None
     if v is not None:
-        assert 2 == 20
+        assert 2 == v
     assert 4 == ray.get(outer)
 
     inner = workflow.get_output("double-2", name="inner")