Improve key error UI