diff --git a/include/evalhyd/detail/probabilist/evaluator.hpp b/include/evalhyd/detail/probabilist/evaluator.hpp
index b4981c6a4f3b2aa225463353c7cb0e58e72c7ab4..0e2ba0561dddc8e7b42c153146ab7c820824e225 100644
--- a/include/evalhyd/detail/probabilist/evaluator.hpp
+++ b/include/evalhyd/detail/probabilist/evaluator.hpp
@@ -91,7 +91,7 @@ namespace evalhyd
             xtl::xoptional<xt::xtensor<double, 6>, bool> CSI;
             xtl::xoptional<xt::xtensor<double, 5>, bool> ROCSS;
             // > Ranks-based
-            xtl::xoptional<xt::xtensor<double, 6>, bool> RANK_DIAG;
+            xtl::xoptional<xt::xtensor<double, 5>, bool> RANK_DIAG;
             xtl::xoptional<xt::xtensor<double, 4>, bool> DS;
             xtl::xoptional<xt::xtensor<double, 4>, bool> AS;
 
@@ -543,7 +543,7 @@ namespace evalhyd
                 return ROCSS.value();
             };
 
-            xt::xtensor<double, 6> get_RANK_DIAG()
+            xt::xtensor<double, 5> get_RANK_DIAG()
             {
                 if (!RANK_DIAG.has_value())
                 {
diff --git a/include/evalhyd/detail/probabilist/ranks.hpp b/include/evalhyd/detail/probabilist/ranks.hpp
index 2efc15fb370100421dcc81f512dd76b95d4e1386..a594b8fac3edfe2089803c550bee0ceeec923104 100644
--- a/include/evalhyd/detail/probabilist/ranks.hpp
+++ b/include/evalhyd/detail/probabilist/ranks.hpp
@@ -208,7 +208,7 @@ namespace evalhyd
             /// \return
             ///     Rank diagram.
             ///     shape: (sites, lead times, subsets, samples, ranks)
-            inline xt::xtensor<double, 6> calc_RANK_DIAG(
+            inline xt::xtensor<double, 5> calc_RANK_DIAG(
                     const xt::xtensor<double, 5>& o_j,
                     const xt::xtensor<bool, 4>& t_msk,
                     const std::vector<xt::xkeep_slice<int>>& b_exp,
@@ -220,9 +220,8 @@ namespace evalhyd
             )
             {
                 // initialise output variable
-                xt::xtensor<double, 6> REL_DIAG =
-                        xt::zeros<double>({n_sit, n_ldt, n_msk, n_exp,
-                                           n_mbr + 1, std::size_t {2}});
+                xt::xtensor<double, 5> REL_DIAG =
+                        xt::zeros<double>({n_sit, n_ldt, n_msk, n_exp, n_mbr + 1});
 
                 // compute variable one mask at a time to minimise memory imprint
                 for (std::size_t m = 0; m < n_msk; m++)