diff --git a/dask_cuda/benchmarks/local_cupy.py b/dask_cuda/benchmarks/local_cupy.py
index 9c6273dee..382721eb4 100644
--- a/dask_cuda/benchmarks/local_cupy.py
+++ b/dask_cuda/benchmarks/local_cupy.py
@@ -75,7 +75,7 @@ async def _run(client, args):
         res = client.compute(func(*func_args))
         await client.gather(res)
         if args.type == "gpu":
-            await client.run(xp.cuda.Device().synchronize)
+            await client.run(lambda xp: xp.cuda.Device().synchronize(), xp)
         took = clock() - t1
 
     return {