Since you haven't done anything yet, you currently don't have anything to enter on 2021 8915-F line 14 (for 401K).
BUT
"Any repayments made before you file your return and by the due date (including extensions) reduce the amount of the distribution included in your income."
