Adjust tests, since now more DistributeNodes are used.