diff --git a/cmrsim/analytic/contrast/_spatial.py b/cmrsim/analytic/contrast/_spatial.py
index 4f352577d0a3dda13fe22f88e13d6b493feba7c2..b89a26517a0a97a261f43e644b3ed7c921266d5b 100644
--- a/cmrsim/analytic/contrast/_spatial.py
+++ b/cmrsim/analytic/contrast/_spatial.py
@@ -250,9 +250,9 @@ class LocalLookREST(BaseSignalModel):
                              f"{orthonormality_bool} \n {orthonormality}")
 
     def update(self):
-        self._validate_shapes(self.orientation_matrix.read_value()[:, 0],
-                              self.orientation_matrix.read_value()[:, 1],
-                              self.orientation_matrix.read_value()[:, 2],
+        self._validate_shapes(self.orientation_matrix.read_value()[:, 0, :3],
+                              self.orientation_matrix.read_value()[:, 1, :3],
+                              self.orientation_matrix.read_value()[:, 2, :3],
                               self.slice_position.read_value(),
                               self.spatial_extend.read_value())
         self.expansion_factor.assign(tf.shape(self.orientation_matrix.read_value())[0])