= Fix Ui.flatMap