Fix bug in decorrelation step