I don't think the timing matters, your 1099R should say 2021. If it doesn't and say 2020 then apply it anyway.. You'll be asked on turbo tax is you had a covid-19 disaster distribution from your 401K distribution for 2019,2020,2021. I believe once you say yes you will then be prompted to select how you want the taxable portion paid back. All at once or spread over 2 or 3 years, 3 being the max. I hope this helps.
