Fix UI in case of errorrs.