diff --git a/tests/func/experiments/test_queue.py b/tests/func/experiments/test_queue.py
index 3618ccfe3..6802da0c7 100644
--- a/tests/func/experiments/test_queue.py
+++ b/tests/func/experiments/test_queue.py
@@ -64,7 +64,7 @@ def test_celery_logs(
     assert "failed to reproduce 'failed-copy-file'" in captured.out
 
 
-@pytest.mark.flaky(rerun=3)
+@pytest.mark.flaky(max_runs=3, min_passes=1)
 def test_queue_remove_done(dvc, failed_tasks, success_tasks):
     assert len(dvc.experiments.celery_queue.failed_stash) == 3
     status = to_dict(dvc.experiments.celery_queue.status())