By entering your K-1 into the program, your state of residence return has already been addressed (i.e., you are taxed on all of your income on your State A return).
The State B return would be a nonresident return for you (if one is required to be filed) and you should prepare that first, if necessary.
