Cancel button fix