Exit button fix