diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py
index b7dee526245c..26d2d5d3ce0e 100644
--- a/tests/python_package_test/test_dask.py
+++ b/tests/python_package_test/test_dask.py
@@ -1656,7 +1656,7 @@ def test_machines_should_be_used_if_provided(task, cluster):
                 dask_model.fit(dX, dy, group=dg)
 
         # The above error leaves a worker waiting
-        client.restart()
+        client.restart(timeout=5, wait_for_workers=True)
 
         # an informative error should be raised if "machines" has duplicates
         one_open_port = lgb.dask._find_n_open_ports(1)