diff --git a/cuda_core/tests/graph/test_graph_node_update.py b/cuda_core/tests/graph/test_graph_node_update.py
index 9f31df6a118..c846f230c26 100644
--- a/cuda_core/tests/graph/test_graph_node_update.py
+++ b/cuda_core/tests/graph/test_graph_node_update.py
@@ -120,7 +120,7 @@ def assert_exec_uses(graph, expected):
         graph.launch(stream)
         try:
             assert callback_started.wait(timeout=5)
-            assert expected.is_done is False
+            _wait_until(lambda: expected.is_done is False)
             unexpected = replacement if expected is original else original
             assert unexpected.is_done is True
         finally: