fixed bug in exact diagonal J with multiple mpi processes