@rvblanton it'll work itself out when you file both a part year resident form for CA and for MT. Since you took the distribution after 7/31, it'll be reported to MT.
