Po pádu železné opony se vrátil do Čech a vstoupil do diplomatických služeb. Stal se velvyslancem v Německu do června roku 1997, poté byl 5 měsíců ministrem školství, mládeže a tělovýchovy. Rok nato se stal velvyslancem v Rakousku a v této funkci vytrval 6 let. Během tohoto období byl vyznamenán Cenou Jaroslava Seiferta za sbírku "Grušas Wacht am Rhein (Grušova Hlídka na Rýnu)", za niž získal i cenu Magnesia Litera. Po roce 2004 se stal ředitelem "Diplomatické akademie Vídeň" a také získal předsednictví v Mezinárodním sdružení PEN-klubů. Obě tyto funkce vykonával do roku 2009, kdy se vrátil k literární tvorbě. Na sklonku života vyšlo jeho dílo "Beneš jako Rakušan". Dne 28. října roku 2011 zemřel během operace srdce ve věku 72 let.