I don't think there's official IRS guidance on that situation. I'm of the opinion that since the money was refunded in 2021 and then spent, on qualified expenses, in the same year (2021), it's still a qualified distribution.
