Fix MPI deadlock in affinity setting