The incarnation was not initially imposed on the spirits as a punishment. It is rather necessary for the spirits' improvement and for the fulfillment of God's works. Everyone has to submit to it, no matter if one takes the evil or the good path. Only those who take the good path will improve quickly, they do not delay to arrive at the end in less painful conditions.