Fix for close button