diff --git a/tests/test_api.py b/tests/test_api.py
index e39327f7..84c03317 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -165,8 +165,6 @@ def test_api_dataset_generator(library, task, model):
 def test_api_launch(benchmark_config, launcher_config, device):
     if launcher_config.name == "torchrun" and device == "cuda":
         device_ids = ",".join(str(i) for i in range(torch.cuda.device_count()))
-    elif device == "cuda":
-        device_ids = "0"
     else:
         device_ids = None