Close button fix