diff --git a/autogalaxy/galaxy/galaxy.py b/autogalaxy/galaxy/galaxy.py
index 7126d4988..63ca30d07 100644
--- a/autogalaxy/galaxy/galaxy.py
+++ b/autogalaxy/galaxy/galaxy.py
@@ -310,6 +310,7 @@ def traced_grid_2d_from(
                 over_sample_size=grid.over_sample_size,
                 over_sampled=grid.over_sampled
                 - self.deflections_yx_2d_from(grid=grid.over_sampled, xp=xp),
+                over_sampler=grid.over_sampler,
             )
 
         return grid - self.deflections_yx_2d_from(grid=grid, xp=xp)