diff --git a/src/Meshers/p4est.jl b/src/Meshers/p4est.jl
index b283160..f70f61c 100644
--- a/src/Meshers/p4est.jl
+++ b/src/Meshers/p4est.jl
@@ -16,7 +16,7 @@ function generate_refined_mesh(
   model = GridapP4est.with(ranks) do
     OctreeDistributedDiscreteModel(ranks,base_model,num_refs)
   end
-  return model
+  return model.dmodel
 end
 
 """
diff --git a/test/mpi/expansion_tests.jl b/test/mpi/expansion_tests.jl
index e9c2a4d..0af8876 100644
--- a/test/mpi/expansion_tests.jl
+++ b/test/mpi/expansion_tests.jl
@@ -2,9 +2,29 @@ module ExpansionTestsMPI
 
 using GridapPETSc
 using SparseMatricesCSR
-
 using GridapMHD: expansion
 
-expansion(np=4,backend=:mpi,savelines=true)
+#expansion(np=4,backend=:mpi,savelines=true)
+
+mesh = Dict{Symbol,Any}(
+  :mesher    => :p4est_SG,
+  :base_mesh => "meshes/Expansion_710.msh",
+  :num_refs  => 1,
+)
+solver = Dict{Symbol,Any}(
+  :solver        => :badia2024,
+  :matrix_type   => SparseMatrixCSR{0,PetscScalar,PetscInt},
+  :vector_type   => Vector{PetscScalar},
+  :block_solvers => [:petsc_mumps,:cg_jacobi,:cg_jacobi],
+  :petsc_options => "-ksp_error_if_not_converged true -ksp_converged_reason",
+)
+expansion(
+  np=4,
+  backend=:mpi,
+  mesh=mesh,
+  solver=solver,
+  ζ  = 10.0,
+  savelines=true
+)
 
 end
\ No newline at end of file