"want to include the total amount on the 2021 tax return"

The way the provision is written:

You either accept the three-year spread, or you opt out and report it all in the year of distribution. There is no provision for changing your mind, mid-way.

There is a provision to Amend 2020, of course.