diff --git a/test/mpi/ThermalComplianceALMTests.jl b/test/mpi/ThermalComplianceALMTests.jl
index e5e6c42..9b7e0a6 100644
--- a/test/mpi/ThermalComplianceALMTests.jl
+++ b/test/mpi/ThermalComplianceALMTests.jl
@@ -193,7 +193,7 @@ with_mpi() do distribute
   @test main(distribute,(2,2);order=1,AD=true)
   @test main(distribute,(2,2);order=2,AD=true)
   @test main(distribute,(2,2);order=1,AD=false)
-  @test main_3d(distribute,(2,2);order=1)
+  @test main_3d(distribute,(2,2,1);order=1)
 end
 
 end # module
\ No newline at end of file
diff --git a/test/mpi/ThermalComplianceALMTestsPETSc.jl b/test/mpi/ThermalComplianceALMTestsPETSc.jl
index febd6a3..db1b6d4 100644
--- a/test/mpi/ThermalComplianceALMTestsPETSc.jl
+++ b/test/mpi/ThermalComplianceALMTestsPETSc.jl
@@ -222,7 +222,7 @@ with_mpi() do distribute
     -ksp_converged_reason -ksp_rtol 1.0e-12"
   GridapPETSc.with(args=split(solver_options)) do
     @test main(distribute,(2,2);order=1,AD=false)
-    @test main_3d(distribute,(2,2);order=1)
+    @test main_3d(distribute,(2,2,1);order=1)
   end
 end