diff --git a/octreelib/grid/grid.py b/octreelib/grid/grid.py
index 84b6d8a..2d3a8fd 100644
--- a/octreelib/grid/grid.py
+++ b/octreelib/grid/grid.py
@@ -161,10 +161,10 @@ def visualize(self, config: VisualizationConfig) -> None:
 
         vertices = []
         for pose_number in range(poses_number):
-            # TODO: Draw full grid, either empty voxels
+            # TODO: Draw full grid including empty voxels
             leaves = self.get_leaf_points(pose_number=pose_number)
             for leaf in leaves:
-                vertices.append(leaf.corners)
+                vertices.append(leaf.all_corners)
 
         for vertex in vertices:
             plot += k3d.lines(