If you took your distribution while you were a resident of Kansas, then you will "Allocate" (assign) that distribution to Kansas.

When you are working on a Part-Year Resident return, the software assumes nothing regarding your income and expenses except in relation to W-2's, which have a state assigned to them.  Therefore, you must manually review the return(s) for accuracy.