diff --git a/tests/tpch/test_optimization.py b/tests/tpch/test_optimization.py
index 851d3d6ab8..28b6deed86 100644
--- a/tests/tpch/test_optimization.py
+++ b/tests/tpch/test_optimization.py
@@ -44,7 +44,6 @@ def test_optimization(query, dataset_path, fs, client, scale):
 
 def test_delay_computation_start(query, dataset_path, fs, client, scale):
     func = getattr(dask_queries, f"query_{query:02d}")
-    result = func(dataset_path, fs, scale).optimize()
     # Client.compute unblocks as soon as update_graph finishes, i.e. graph is
     # submitted and parsed. This is the time until the dashboard kicks off
-    client.compute(result)
+    client.compute(func(dataset_path, fs, scale))