Retirement tax questions

My hope was to file 2020 return now reflecting 1/3 repayment to original 401k & make the actual 1/3 repayment l just prior to May 17 filing deadline. 

Hope that is OK