Covid distribution from 401k

We withdrew a 401k in august of 2020.

I haven’t filed my 2022 taxes but if I make a payment this week in 2023 will it be applied to next tax year or do I amend my 2022 return to show a repayment? 

or can I amend a 2020 return to elect it all be claimed at once instead of spilt. 
thanks