If you complete your CA return before you do your non-resident AZ return, the program should do this calculation for you.
