Amend 2021.

And, amend the shareholders personal returns. Or just issue the corrected K-1s....

Yes, you have to amend 2021 reducing the deduction for wages paid by the amount of the Employee Retention Credit.  This will change the K-1s.

You can pick up any interest the IRS pays on top of the tax refund in the current year.