Lund gikk av som leder for FO II i 1946. Han ble da militærrådgiver for FNs første generalsekretær, nordmannen Trygve Lie. Da Folke Bernadotte ble FN sin fredsmegler i Palestina ble Lund Bernadottes midlertidige stabssjef. Etter at han ikke ble akseptert av Sovjetunionen som militærattache i Moskva arbeidet han i Forsvarsdepartementets transportkontor.