diff --git a/autoarray/structures/grids/uniform_2d.py b/autoarray/structures/grids/uniform_2d.py
index 80bccfe7..d755951d 100644
--- a/autoarray/structures/grids/uniform_2d.py
+++ b/autoarray/structures/grids/uniform_2d.py
@@ -171,6 +171,9 @@ def __init__(
 
         grid_2d_util.check_grid_2d(grid_2d=values)
 
+        if os.environ.get("PYAUTO_WORKSPACE_SMALL_DATASETS") == "1":
+            over_sample_size = 2
+
         self._over_sample_size = over_sample_size
 
         self._over_sampler = over_sampler