Fix cancel button issue

whooooooooo