Did you read the sections referred to in this document:

https://www.irs.gov/pub/irs-drop/n-05-92.pdf

You have to know if your taxpayer elected to report the distribution entirely in 2020, or let it be spread over the three years, as well. The repayments are essentially rollovers, so you should get the 5498 for each year of repayment, to confirm.