diff --git a/keras/src/ops/image_test.py b/keras/src/ops/image_test.py
index bcf42b533a8..7f346abca96 100644
--- a/keras/src/ops/image_test.py
+++ b/keras/src/ops/image_test.py
@@ -351,7 +351,7 @@ def _compute_affine_transform_coordinates(image, transform):
     # transform the indices
     coordinates = np.einsum("Bhwij, Bjk -> Bhwik", indices, transform)
     coordinates = np.moveaxis(coordinates, source=-1, destination=1)
-    coordinates += np.reshape(a=offset, newshape=(*offset.shape, 1, 1, 1))
+    coordinates += np.reshape(offset, newshape=(*offset.shape, 1, 1, 1))
     if need_squeeze:
         coordinates = np.squeeze(coordinates, axis=0)
     return coordinates