Retirement tax questions

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

Hope that is OK