diff --git a/observation_sim/mock_objects/FlatLED.py b/observation_sim/mock_objects/FlatLED.py
index bfa58ff82f0e2fce1975f9b5a29aeaa10759d600..2674aa5c6ff8659db1c06a96e070d74db9c61d7d 100755
--- a/observation_sim/mock_objects/FlatLED.py
+++ b/observation_sim/mock_objects/FlatLED.py
@@ -150,10 +150,7 @@ class FlatLED(MockObject):
         U = np.zeros([self.chip.npix_y, self.chip.npix_x], dtype=np.float32)
         for y_seg_i in np.arange(y_seg_len):
             for x_seg_i in np.arange(x_seg_len):
-                U[y_seg_i*y_seg:(y_seg_i+1)*y_seg, x_seg_i*x_seg:(x_seg_i+1)*x_seg] = griddata(X_, Z_, (
-                M[y_seg_i*y_seg:(y_seg_i+1)*y_seg, x_seg_i*x_seg:(x_seg_i+1)*x_seg],
-                N[y_seg_i*y_seg:(y_seg_i+1)*y_seg, x_seg_i*x_seg:(x_seg_i+1)*x_seg]),
-                method='linear')
+                U[y_seg_i*y_seg:(y_seg_i+1)*y_seg, x_seg_i*x_seg:(x_seg_i+1)*x_seg] = griddata(X_, Z_, (M[y_seg_i*y_seg:(y_seg_i+1)*y_seg, x_seg_i*x_seg:(x_seg_i+1)*x_seg], N[y_seg_i*y_seg:(y_seg_i+1)*y_seg, x_seg_i*x_seg:(x_seg_i+1)*x_seg]), method='linear')
         # U = griddata(X_, Z_, (
         #     M[0:self.chip.npix_y, 0:self.chip.npix_x],
         #     N[0:self.chip.npix_y, 0:self.chip.npix_x]),