More bug fixes in error UI