Screen 32 > For each partner there is an input Pass - Through Entity Elective Tax Calculation (Form 3804) with a yes/no and pro-rata [O] I think check this box as a yes (1) will trigger the calculation

I really have not played with how the program does this. Was the tax prepaid in 2021?

You may need to do a special allocation.