diff --git a/extensions/chaeo/products.py b/extensions/chaeo/products.py
index d0ec388a90d9e07b9082a060e67afc6acc1ac6e0..b056a4ad23222ba2bccea374c3371d5cc2ef8491 100644
--- a/extensions/chaeo/products.py
+++ b/extensions/chaeo/products.py
@@ -113,11 +113,11 @@ def export_3d_patches_with_focus_metrics(
             'max_sobel': lambda x: np.max(sobel(x)),
             'rms_sobel': lambda x: sqrt(np.mean(sobel(x) ** 2)),
             'entropy': lambda x: shannon_entropy(x),
-            'moment': lambda x: moment(x, moment=2),
+            'moment': lambda x: moment(x.flatten(), moment=2),
         }
         dd = {}
         for zi in range(0, nz):
-            spf = zs[:, :, :, zi].flatten()
+            spf = zs[:, :, :, zi]
             dd[zi] = {k: me[k](spf) for k in me.keys()}
         return dd