А нельзя ли слушать в attach событие focus на button и фокусить нужный элемент в обработчике?