Fix dialog button focus behavior