diff --git a/src/iterative_ensemble_smoother/experimental.py b/src/iterative_ensemble_smoother/experimental.py
index 5db70166..3806cb87 100644
--- a/src/iterative_ensemble_smoother/experimental.py
+++ b/src/iterative_ensemble_smoother/experimental.py
@@ -313,7 +313,7 @@ def progress_callback(x):
 
         if correlation_callback is not None:
             corr_XY = self._cov_to_corr_inplace(cov_XY, stds_X, stds_Y)
-            correlation_callback(corr_XY[significant_rows])
+            correlation_callback(corr_XY)
         return X