diff --git a/tests/test_distributed.py b/tests/test_distributed.py
index 40eba09..659680d 100644
--- a/tests/test_distributed.py
+++ b/tests/test_distributed.py
@@ -40,18 +40,21 @@ def test_run_distributed(args):
         if core_fraction > 0:
             pytest.raises(
                 ZeroDivisionError,
-                call_wrapper(args),
+                call_wrapper,
+                args=(args,),
                 match="division by zero",
             )
         else:
             pytest.raises(
                 ModuleNotFoundError,
-                call_wrapper(args),
+                call_wrapper,
+                args=(args,),
                 match="Currently dask.distrubuted is not installed",
             )
     else:
         pytest.raises(
             ModuleNotFoundError,
-            call_wrapper(args),
+            call_wrapper,
+            args=(args,),
             match="Currently dask.distrubuted is not installed",
         )