Add missing callback for click on close button