Fix calls to ui.flash