LDA - Fix a data redistribution issue