diff --git a/test/mpi/OctreeDistributedDiscreteModelsNoEnvTests.jl b/test/mpi/OctreeDistributedDiscreteModelsNoEnvTests.jl
index cfd7acd..d81fde6 100644
--- a/test/mpi/OctreeDistributedDiscreteModelsNoEnvTests.jl
+++ b/test/mpi/OctreeDistributedDiscreteModelsNoEnvTests.jl
@@ -7,8 +7,14 @@ import .OctreeDistributedDiscreteModelsTests as TestModule
 
 MPI.Init()
 
+# 2D 
 with_mpi() do distribute
-  TestModule.run(distribute,(2,2),[4,2])
-end 
+  TestModule.run(distribute,4,(2,2),[4,2])
+end
+
+# 3D
+with_mpi() do distribute
+  TestModule.run(distribute,4,(2,2,2),[4,2])
+end
 
 MPI.Finalize()