Da det blev klart for ham, at Tyskland var ved at tabe krigen, rejste han til Stockholm, hvor han forsøgte at indlede fredsforhandlinger. Han blev arresteret i juni 1945 og reddede sig selv fra en lang straf ved at vidne mod en række nazister ved Nürnbergprocessen. Under sin fængsling skrev han den selvbiografiske bog "The Labyrinth".