fix close UX