Add error handling to dialogs