fix close button bug