diff --git a/maze.py b/maze.py
index 981b895..187b10d 100644
--- a/maze.py
+++ b/maze.py
@@ -121,7 +121,10 @@ class Maze:
         else:
             self._graphics.clear_all()
             self._reset_cell_grid()
-        self._draw_cell_grid()
+
+        if self._graphics:
+            self._draw_cell_grid()
+
         self._open_entrance_and_exit()
         self._break_walls_r(MazePosition(
             i=0,
diff --git a/tests.py b/tests.py
index b3a1a3c..b6d59b8 100644
--- a/tests.py
+++ b/tests.py
@@ -39,6 +39,7 @@ class Tests(unittest.TestCase):
                 None,
                 None,
             )
+            m.generate()
             self.assertEqual(
                 len(m._cell_grid),
                 case["height"],