Retirement tax questions

Since you repaid the money in 2021, you don't owe the taxes.

 

You could have delayed filing your 2020 return so that you could report the repayment.  Since you did file a 2020 return and pay the taxes, you will need to file an amended 2020 return to report the repayment and get the taxes back.