You pay the tax on the distribution in 2020 (with the option to spread the tax out over 3 years).
Then if you really do pay back the money, you would file an amended 2020 tax return to report the payback and recalculate your tax, resulting in a refund on the amended return.
