Modified test so that only 2 PEs participate in the collect.