After you file

If they took out withholding for the wrong state you have to file a non resident state return for that state to get it back.  You will probably owe tax to the right state.