To make the question about a 'disaster distribution' pop up, you need to enter a 1099-R in the Retirement section under Wages/Income.
If you elected to spread your distribution over three years, enter the amount as one-third of the amount on your original 1099-R.
You should then come to a screen asking for info from your 2021 return.
Here's more info on Entering Disaster Distributions.