fix UI for success+error