Fix bug with NoDist dimensions and redistribution.