diff --git a/deepinv_bench/benchmarks/div2k_gaussian_deblurring/solvers/dps.py b/deepinv_bench/benchmarks/div2k_gaussian_deblurring/solvers/dps.py
index 8486eac..1ef305b 100644
--- a/deepinv_bench/benchmarks/div2k_gaussian_deblurring/solvers/dps.py
+++ b/deepinv_bench/benchmarks/div2k_gaussian_deblurring/solvers/dps.py
@@ -30,7 +30,7 @@ def set_objective(self, train_dataset=None, physics=None):
             raise NotImplementedError
 
         self.model = dinv.sampling.DPS(
-            model=denoiser, device=device, max_iter=self.max_iter
+            denoiser=denoiser, device=device, max_iter=self.max_iter
         )
         self.model.device = device
 
diff --git a/deepinv_bench/benchmarks/div2k_inpainting_easy/solvers/dps.py b/deepinv_bench/benchmarks/div2k_inpainting_easy/solvers/dps.py
index 8486eac..1ef305b 100644
--- a/deepinv_bench/benchmarks/div2k_inpainting_easy/solvers/dps.py
+++ b/deepinv_bench/benchmarks/div2k_inpainting_easy/solvers/dps.py
@@ -30,7 +30,7 @@ def set_objective(self, train_dataset=None, physics=None):
             raise NotImplementedError
 
         self.model = dinv.sampling.DPS(
-            model=denoiser, device=device, max_iter=self.max_iter
+            denoiser=denoiser, device=device, max_iter=self.max_iter
         )
         self.model.device = device
 
diff --git a/deepinv_bench/benchmarks/div2k_super_resolution_2x/solvers/dps.py b/deepinv_bench/benchmarks/div2k_super_resolution_2x/solvers/dps.py
index 8486eac..1ef305b 100644
--- a/deepinv_bench/benchmarks/div2k_super_resolution_2x/solvers/dps.py
+++ b/deepinv_bench/benchmarks/div2k_super_resolution_2x/solvers/dps.py
@@ -30,7 +30,7 @@ def set_objective(self, train_dataset=None, physics=None):
             raise NotImplementedError
 
         self.model = dinv.sampling.DPS(
-            model=denoiser, device=device, max_iter=self.max_iter
+            denoiser=denoiser, device=device, max_iter=self.max_iter
         )
         self.model.device = device