diff --git a/autolens/plot/__init__.py b/autolens/plot/__init__.py
index 57dd8f2a1..94b640250 100644
--- a/autolens/plot/__init__.py
+++ b/autolens/plot/__init__.py
@@ -28,7 +28,6 @@
     GridPlot,
     VectorYXQuiver,
     PatchOverlay,
-    DelaunayDrawer,
     VoronoiDrawer,
     OriginScatter,
     MaskScatter,
diff --git a/autolens/point/solver/shape_solver.py b/autolens/point/solver/shape_solver.py
index f85043add..f398f41cd 100644
--- a/autolens/point/solver/shape_solver.py
+++ b/autolens/point/solver/shape_solver.py
@@ -12,22 +12,17 @@
     if use_jax:
         from autoarray.structures.triangles.coordinate_array.jax_coordinate_array import (
             CoordinateArrayTriangles,
-            MAX_CONTAINING_SIZE,
         )
     else:
         from autoarray.structures.triangles.coordinate_array.coordinate_array import (
             CoordinateArrayTriangles,
         )
 
-        MAX_CONTAINING_SIZE = None
-
 except ImportError:
     from autoarray.structures.triangles.coordinate_array.coordinate_array import (
         CoordinateArrayTriangles,
     )
 
-    MAX_CONTAINING_SIZE = None
-
 from autoarray.structures.triangles.abstract import AbstractTriangles
 
 from autogalaxy import OperateDeflections
@@ -76,7 +71,6 @@ def for_grid(
         pixel_scale_precision: float,
         magnification_threshold=0.1,
         array_triangles_cls: Type[AbstractTriangles] = CoordinateArrayTriangles,
-        max_containing_size=MAX_CONTAINING_SIZE,
         neighbor_degree: int = 1,
     ):
         """
@@ -124,7 +118,6 @@ def for_grid(
             pixel_scale_precision=pixel_scale_precision,
             magnification_threshold=magnification_threshold,
             array_triangles_cls=array_triangles_cls,
-            max_containing_size=max_containing_size,
             neighbor_degree=neighbor_degree,
         )
 
@@ -139,7 +132,6 @@ def for_limits_and_scale(
         pixel_scale_precision: float = 0.001,
         magnification_threshold=0.1,
         array_triangles_cls: Type[AbstractTriangles] = CoordinateArrayTriangles,
-        max_containing_size=MAX_CONTAINING_SIZE,
         neighbor_degree: int = 1,
     ):
         """
@@ -163,9 +155,6 @@ def for_limits_and_scale(
         array_triangles_cls
             The class to use for the triangles. JAX is used implicitly if USE_JAX=1 and
             jax is installed.
-        max_containing_size
-            Only applies to JAX. This is the maximum number of multiple images expected.
-            We need to know this in advance to allocate memory for the JAX array.
         neighbor_degree
             The number of times recursively add neighbors for the triangles that contain
 
@@ -179,7 +168,6 @@ def for_limits_and_scale(
             x_min=x_min,
             x_max=x_max,
             scale=scale,
-            max_containing_size=max_containing_size,
         )
 
         return cls(