Fixes MPI data sent when A is lower.