1 files changed, 3 insertions, 2 deletions
diff --git a/CLI/rust/src/game_of_life.rs b/CLI/rust/src/game_of_life.rs
index 6b52034..96ec3d9 100644
--- a/CLI/rust/src/game_of_life.rs
+++ b/CLI/rust/src/game_of_life.rs
@@ -175,12 +175,13 @@ pub(crate) fn generate(opts: Opts) -> Result<Universe, Error> {
 mod generate {
     use super::*;
     use mocktopus::mocking::*;
+    use crate::game_of_life::universe::Plane;
 
     fn universe_mock(opts: Opts) -> Universe {
         Universe {
             opts,
-            current: vec![vec![]],
-            future: vec![vec![]],
+            current: vec![Plane(vec![])],
+            future: vec![Plane(vec![])],
         }
     }