Paid back 2/3's of money withdrawn from 401K for covid hardship withdrawl in 2020. How do I represent this for year 2021 taxes so I no longer owe taxes on the 1/3 from 2021?

Hello,

I took out a Covid Hardship withdrawl from my 401K in 2020 and elected to divide the tax liability over tax years 2020, 2021, and 2022. I jpaid back 2/3 of this distribution in 2022 and now would like to amend my 2021 tax return to remove the federal and state taxes and penalties that I owe. How do I amend and refile my 2021 tax return to indicate that I have repaid this distribution so my tax liability and penalties will be removed for that year?  

Thank you