diff --git a/tests/test_models.py b/tests/test_models.py
index 62963e2c..f217bf18 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -339,7 +339,7 @@ def test_running_test_pretrained_model_dp():
         val_percent_check=0.2,
         checkpoint_callback=checkpoint,
         experiment=exp,
-        gpus=2,
+        gpus=[0, 1],
         distributed_backend='dp'
     )
 
@@ -1247,7 +1247,7 @@ def test_amp_gpu_ddp_slurm_managed():
     trainer_options = dict(
         show_progress_bar=True,
         max_nb_epochs=1,
-        gpus=2,
+        gpus=1,
         distributed_backend='ddp',
         use_amp=True
     )