Walt doesn't have to die for there to be justice. Based on the last episode, he's already lost his family and most of his money, and word's going to get out that he's Heisenberg. Everything he cares about is gone. Whether he comes back to save Jesse or not, he's paying for all his crimes even if he doesn't get killed. A long, miserable life is a more fitting punishment than a swift death, in any case.