fix assign UI button