@AmeliesUncle is correct ... you cannot amend 2020 & 2021 to pay the taxes on the 2020 return only once you have made the election. Your only option would be to amend the 2020 & 2021 returns ONLY IF you pay back the entire distribution to negate the original distribution which will probably hurt much more in the long run ... but if you could repay some or all of the distribution then you may have an option.
