Fixed issue with close button :ant: