a repayment has to be recorded by the custodian as such.
meanwhile your 401k payroll contributions continue if you are employed.
you have three years to pay back some or all and get a refund, including for year 2020 (if done by May 17, or Oct 15 with an extension).
talk to your custodian about how to do it.
