diff --git a/core/distributed/preconditioner/schwarz.cpp b/core/distributed/preconditioner/schwarz.cpp
index c9df112b8b9..abde203895c 100644
--- a/core/distributed/preconditioner/schwarz.cpp
+++ b/core/distributed/preconditioner/schwarz.cpp
@@ -58,8 +58,7 @@ void Schwarz<ValueType, LocalIndexType, GlobalIndexType>::apply_dense_impl(
 
         restrict->apply(dense_b, this->csol_);
         this->coarse_solver_->apply(this->csol_, this->csol_);
-        prolong->apply(this->half_.get(), this->csol_.get(), this->half_.get(),
-                       dense_x);
+        prolong->apply(this->half_, this->csol_, this->half_, dense_x);
     }
 }
 
diff --git a/examples/distributed-solver/distributed-solver.cpp b/examples/distributed-solver/distributed-solver.cpp
index dd2f102b738..acda7a52f8c 100644
--- a/examples/distributed-solver/distributed-solver.cpp
+++ b/examples/distributed-solver/distributed-solver.cpp
@@ -211,7 +211,7 @@ int main(int argc, char* argv[])
             solver::build()
                 .with_preconditioner(
                     schwarz::build()
-                        .with_local_solver_factory(local_solver)
+                        .with_local_solver(local_solver)
                         .with_galerkin_ops_factory(pgm_fac)
                         .with_coarse_solver_factory(coarse_solver)
                         .on(exec))
@@ -228,11 +228,7 @@ int main(int argc, char* argv[])
         Ainv =
             solver::build()
                 .with_preconditioner(
-                    schwarz::build()
-                        .with_local_solver_factory(local_solver)
-                        .with_galerkin_ops_factory(pgm_fac)
-                        .with_coarse_solver_factory(coarse_solver)
-                        .on(exec))
+                    schwarz::build().with_local_solver(local_solver).on(exec))
                 .with_criteria(
                     gko::stop::Iteration::build().with_max_iters(num_iters).on(
                         exec),