Bug fix: sample assignment in pt.mpi