On the 1099R worksheet, you can choose entire amount, or you can enter an amount....cant you just input the 50k there?    

That means in 2020 he'd pay tax on 45k + 16,667 (1/3 of the 50k), then only 16,666 in 2021 and 16,666 in 2022.

Is that not how it works out?