Improve key assign focus managing.