Do details and click the +  or the 3 verticle dots at the taxable amount and the gross

Then you can get a grid like this to split between the states

This article may help

https://proconnect.intuit.com/support/en-us/help-article/multi-state-taxes/entering-multi-state-amou...

Was the distribution made entirely during CA residency?  If so, it shouldn't be taxable to AZ.