Retirement tax questions

if you received $40,000 and

• deposited $38,000 into an IRA or 401k and designated it as a "repayment of COVID-19 distribution"
• Filed Form 8915-E correctly,

Then you would be correct in your thinking and would not have an issue with the IRS.