Update the epoch stake in use in wen_restart

[wen-coding]

In the current wen_restart implementation, we use epoch_stakes from the root bank, which is wrong if some validators have root bank in epoch X while others have root bank in epoch X+1, remember to fix it later.