Slightly better UI error handling