Fix modal UI