Restore the IME language status after you switch back to the edit box, so that it won't change after pressing Shift.