fix : Ui controller