diff --git a/openwfs/algorithms/dual_reference.py b/openwfs/algorithms/dual_reference.py
index a16d2f7..fd87d32 100644
--- a/openwfs/algorithms/dual_reference.py
+++ b/openwfs/algorithms/dual_reference.py
@@ -126,7 +126,7 @@ def amplitude(self, value):
                 (np.ones(shape=self._shape) / np.sqrt(self.masks[side].sum())).astype(np.float32) for side in range(2))
             return
 
-        if value.shape != self._shape:
+        if value[0].shape != self._shape or value[1].shape != self._shape :
             raise ValueError(
                 "The amplitude and group mask must all have the same shape."
             )