1099R Covid distribution payback

in 2020 we took a covid distribution from the IRA reported on 1099R.  When I input to TT, it asks if any was paid back.  on 8915E, it sounds like you can pay back some of the amount as long as you do before you file the taxes which would be paying now in 2021.  TT then reduces your tax amount owed.  Is this the way the law states that it can be done? i would then get a 5498 showing the amount paid pack.